洛谷 P1564 膜拜

题目出处

s[i]表示前i个人对神牛的膜拜情况，如果膜拜神牛甲则s[i]=s[i-1]+1否则s[i]=s[i-1]-1。那么如果|s[i]-s[j]|<=m或者=i-j+1（也就是人数差不超过m或者全部崇拜某一个神牛），f[i]=min(f[i],f[i-j]+1)

下放代码

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<cmath>
 using namespace std;
 const int maxn=;
 int n,m,f[maxn],s[maxn],a;
 int main(){
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++){
         scanf("%d",&a);
         if(a==)s[i]=s[i-]+;
         else s[i]=s[i-]-;
     }
     memset(f,,sizeof(f));
     f[]=;
     for(int i=;i<=n;i++)
         for(int j=;j<i;j++){
             int c=abs(j?(s[i]-s[j-]):(s[i]));
             if(c==i-j+||c<=m)
                 f[i]=min(f[i],f[j-]+);
         }
     printf("%d",f[n]);
     return ;
 }