hdu 6376 度度熊剪纸条

度度熊剪纸条

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1379    Accepted Submission(s): 202
 

Problem Description

度度熊有一张纸条和一把剪刀。

纸条上依次写着 N 个数字，数字只可能是 0 或者 1。

度度熊想在纸条上剪 K 刀（每一刀只能剪在数字和数字之间），这样就形成了 K+1 段。

他再把这 K+1 段按一定的顺序重新拼起来。

不同的剪和接的方案，可能会得到不同的结果。

度度熊好奇的是，前缀 1 的数量最多能是多少。

Input

有多组数据，读到EOF结束。

对于每一组数据，第一行读入两个数 N 和 K 。

第二行有一个长度为 N 的字符串，依次表示初始时纸条上的 N 个数。

0≤K<N≤10000

所有数据 N 的总和不超过100000

Output

对于每一组数据，输出一个数，表示可能的最大前缀 1 的数量。

Sample Inpu

5 1

11010

5 2

11010

 

Sample Output

2 3

 贪心，将每一段1记录需要切割次数，注意到在选段中可以有一个后端不剪。

#include <bits/stdc++.h>
using namespace std;
#define N 305100
#define ll long long
#define INF 0x3f3f3f3f
struct node {
    int l, r;
    int sum;
    int flag;
} a[N];
int flag;
char s[N];
bool cmp(node a, node b) {
    return a.sum > b.sum;
}
int main () {
    int T;
    int ans;
    int n, k;
    while(scanf("%d%d", &n, &k) != EOF) {
        scanf("%s", s);
        int len = strlen(s);
        flag = 0;
        int cnt = 0, sum = 0, res[4];
        int L, R;
        ans = 0;
        for(int i = 0; i < len; ++i) {
            if(flag == 0&&s[i] == '1') {
                if(i == 0) {
                    a[cnt].flag = 1;
                    a[cnt].l = 0;
                } else {
                    a[cnt].flag = 0;
                    a[cnt].l = 1;
                }
                flag = 1;
                L = i;
            } else if(flag == 1&&s[i]=='0') {
                a[cnt].r = 1;
                a[cnt++].sum = i-L;
                a[cnt].flag = 0;
                flag = 0;
            }
        }
        if(s[len-1] == '1') {
            if(a[cnt].flag == 0)    //注意全为1的情况
                a[cnt].flag = 2;
            a[cnt].r = 0;
            a[cnt++].sum = len-L;
        }
        sort(a, a + cnt, cmp);
        /* for(int i = 0; i < cnt; ++i) {
             printf("%d ", a[i].sum);
         }*/
        //cout << cnt << endl;
        int i;
        for(i = 0; i < cnt; ++i) {
            if(sum >= k)
                break;
            sum += a[i].l+a[i].r;
            ans += a[i].sum;
        }
        int maxi = a[i-1].sum;
        int tot = 0;
        if(k == 0) {
            for(; i < cnt; ++i) {
                if(a[i].flag==1) {
                    ans += a[i].sum;
                    break;
                }
            }
        } else {
            for(; i < cnt; ++i) {
                if(a[i].flag) {
                    res[tot++] = a[i].sum;
                }
            }
            if(tot == 1 ) {
                if(sum <= k)    //去掉一个r
                    ans += res[0];
            }else if(tot >= 2){
                int temp = 0;
                if(sum <= k)    //去掉一个r
                    temp = ans + res[0];
                ans = max(ans-maxi+max(maxi, res[0]+res[1]), temp); //与淘汰最后一个+前端后端作比较
            }
        }

        printf("%d\n", ans);
    }
    return 0;
}
/*
7 4
0101101
7 1
110111011
7 2
1111101111111011111111
1 4
0010
1 0
1
*/