洛谷 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=2505;
int n,m,f[maxn],s[maxn],a;
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a);
        if(a==1)s[i]=s[i-1]+1;
        else s[i]=s[i-1]-1;
    }
    memset(f,101,sizeof(f));
    f[1]=1;
    for(int i=2;i<=n;i++)
        for(int j=0;j<i;j++){
            int c=abs(j?(s[i]-s[j-1]):(s[i]));
            if(c==i-j+1||c<=m)
                f[i]=min(f[i],f[j-1]+1);
        }
    printf("%d",f[n]);
    return 0;
}