你发现染 $k$ 次最多会将这个序列分成 $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;
}