题意
众所周知,白神是具有神奇的能力的。
比如说,他对数学作业说一声“数”,数学作业就会出于畏惧而自己完成;对语文作业说一声“语”,语文作业就会出于畏惧而自己完成。
今天,语文老师和数学老师布置了许多作业,同学们纷纷寻找白神寻求帮助。白神作为一个助人为乐的人,便答应下来。
回到家,白神将这N份作业按顺序摊开,发现语文作业数学作业混在一起,这就让白神苦恼起来,他如果对连续一段作业喊出“数”,那么里面的语文作业就会由于过于慌乱而写满错解,不过如果白神再对其喊一声“语”,它又会写满正确答案。
虽然白神很强大,但是能力还是有限制的,一天只能使用K次,现在,白神想知道他能正确的完成多少份作业。
数据范围
100%的数据中N ≤ 100000,K<=50.
题解
我们可以发现,若可以用k次能力,那么我们最多可以完全覆盖(正确完成)的区间是:
一段0-一段1-一段0-…-一段1-一段0
最多是2*k-1段。
这样理解:每次我们选出最大范围的区间,在这个区间里选一段区间,不断的在选中的区间里选,知道用完k次,可以得出,可以分成2*k-1个区间。
设
表示前i个数分成j段且当前数字为0或1(第三维表示)的最优解。
我们一波循环数组就装下了。
代码
#include<bits/stdc++.h>
using namespace std;
const int N=112345;
int n,k,t[N],ans,f[2][105][2];
int x,y;
inline int rd()
{
char ch=getchar();int x=0,f=1;
while(ch<'0' || ch>'9'){if(ch=='-') f=-1;ch=getchar();}
while(ch>='0' && ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
return x*f;
}
int main(){
n=rd();k=rd();
for(int i=1;i<=n;i++) t[i]=rd();
f[0][1][t[1]]=1;
x=0,y=1;k<<=1;
for(int j,a,b,i=1;i<=n-1;i++,swap(x,y)){
for(j=1;j<k;j++){
for(a=0;a<2;a++){
for(b=0;b<2;b++){
f[y][j+(a!=b)][b]=max(f[y][j+(a!=b)][b],f[x][j][a]+(t[i+1]==b));
}
}
}
for(j=1;j<k;j++){
f[x][j][0]=f[x][j][1]=0;
ans=max(ans,max(f[y][j][0],f[y][j][1]));
}
}
printf("%d\n",ans);
return 0;
}