2018 “百度之星”程序设计大赛 - 初赛（A） 度度熊剪纸条 （01背包）

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;
}