度度熊剪纸条
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
*/