http://acm.hdu.edu.cn/showproblem.php?pid=6376
思路:
先把连续的1串处理出来,如果是开头的串的不需要剪,中间的串需要剪两次,最后面的串只需剪1次,然后枚举,把其他接在开头当做前缀的情况,或者把其他接在中间最大当做前缀的情况,然后把其他接在最后当做最大前缀的情况,比较哪次最优即可。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e4+10;
const int mod=998244353;
int n,k;
struct node
{
int t,len;
}a[maxn];
int cnt;
char c[maxn];
int dp[maxn];
int work(int k)
{
memset(dp,0,sizeof(dp));
for(int i=0;i<cnt;i++)
{
for(int j=k;j>=a[i].t;j--)
{
dp[j]=max(dp[j],dp[j-a[i].t]+a[i].len);
}
}
return dp[k];
}
int main()
{
while(~scanf("%d%d",&n,&k))
{
scanf("%s",c+1);
int f=0;
int len=0;
cnt=0;
int ma=0,id=-1,z=-1;
for(int i=1;i<=n;i++)
{
if(c[i]=='1')
len++;
else
{
if(f==0)
{
a[cnt].len=len;
a[cnt++].t=0;
if(ma<len)
{
ma=len;
id=cnt-1;
z=0;
}
}
else
{
a[cnt].len=len;
a[cnt++].t=2;
if(ma<len)
{
ma=len;
id=cnt-1;
z=2;
}
}
len=0;
f=1;
}
}
a[cnt].len=len;
a[cnt++].t=1;
int ans=0;
if(k==0)
{
printf("%d\n",a[0].len);
continue;
}
if(id!=-1)
{
a[0].t=1;
a[id].t=0;
ans=max(ans,work(k-1));
a[id].t=z;
a[0].t=0;
}
ans=max(ans,work(k));
if(!a[cnt-1].len)
{
a[cnt-1].t=0;
a[0].t=1;
ans=max(ans,work(k-1));
}
printf("%d\n",ans);
}
return 0;
}