BZOJ3791 作业 动态规划

你发现染 $k$ 次最多会将这个序列分成 $2k-1$ 段，然后任何 $2k-1$ 段以内的方案一定能被构建出来，所以直接 dp 就好了 

#include <bits/stdc++.h>
#define N 100004 
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;    
int A[N],f[N][104][2];        
int main() 
{
    int n,k,i,j;
    // setIO("input");  
    scanf("%d%d",&n,&k);         
    k=k*2-1;      
    for(i=1;i<=n;++i) scanf("%d",&A[i]);   
    int ans=0,p; 
    for(i=1;i<=n;++i) 
    {
        for(j=1;j<=k;++j) 
        {
            if(A[i]==0) 
            {
                f[i][j][0]=max(f[i-1][j-1][1]+1, f[i-1][j][0]+1); 
                f[i][j][1]=f[i-1][j][1];   
            }    
            else 
            {
                f[i][j][0]=f[i-1][j][0]; 
                f[i][j][1]=max(f[i-1][j-1][0]+1, f[i-1][j][1]+1);                    
            }
            ans=max(ans, f[i][j][0]); 
            ans=max(ans, f[i][j][1]); 
        }
    } 
    printf("%d\n",ans);    
    return 0;         
}