HDU 6376(思维+01背包)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6376

 

题目大意：给一个长度为n的串，剪k次，问能形成的最大前缀1能有多长

 

题目思路：每一段1的长度为每一段的价值，花费需要特殊判断，中间的直接2，需要剪两下，两边的需要看最后一个和第一个是不是1，如果第一个是1那最左边的花费是1，最后一个是1那最后的花费是1。然后总钱数应该是k+1，因为如果拿最左边的当最大前缀1的尾巴那根本不用剪，但是花费是1，如果拿中间的一段当最后一段，那也不需要剪两下，剪一下就行。末尾是1结尾的话也不用剪，所以他给的次数是k次，但实际上我们用到的花费是k+1，多算了一下，然后直接跑01背包就行。

 

以下是代码：

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define MAXN 10005
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
char s[MAXN];
struct node{
    int val,cost;
}a[MAXN];
int dp[MAXN];
int main(){
    int n,k;
    while(~scanf("%d%d",&n,&k)){
        scanf("%s",s+1);
        int tot=0,flag=0,num=0;
        rep(i,1,n){
            if(s[i]=='0'&&flag){
                flag=0;
                a[++tot].val=num;
                if(tot==1&&s[1]=='1')a[tot].cost=1;
                else a[tot].cost=2;
                num=0;
            }
            if(s[i]=='1'){
                flag=1;
                num++;
            }
        }
        if(flag){
            a[++tot].val=num;a[tot].cost=1;
        }
        if(k==0){
            if(s[1]=='1')printf("%d\n",a[1].val);
            else printf("0\n");
        }
        else{
            k++;
            memset(dp,0,sizeof(dp));
            rep(i,1,tot){
                per(j,k,a[i].cost){
                    dp[j]=max(dp[j],dp[j-a[i].cost]+a[i].val);
                }
            }
            printf("%d\n",dp[k]);
        }
    }
    return 0;
}