洛谷 P1564 膜拜

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 #include<cmath>
 5 using namespace std;
 6 const int maxn=2505;
 7 int n,m,f[maxn],s[maxn],a;
 8 int main(){
 9     scanf("%d%d",&n,&m);
10     for(int i=1;i<=n;i++){
11         scanf("%d",&a);
12         if(a==1)s[i]=s[i-1]+1;
13         else s[i]=s[i-1]-1;
14     }
15     memset(f,101,sizeof(f));
16     f[1]=1;
17     for(int i=2;i<=n;i++)
18         for(int j=0;j<i;j++){
19             int c=abs(j?(s[i]-s[j-1]):(s[i]));
20             if(c==i-j+1||c<=m)
21                 f[i]=min(f[i],f[j-1]+1);
22         }
23     printf("%d",f[n]);
24     return 0;
25 }