度度熊剪纸条
Accepts: 488
Submissions: 5471
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 65536/65536 K (Java/Others)
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 Input
Copy
5 1 11010 5 2 11010
Sample Output
Copy
2 3
贪心 分类讨论
#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<string>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
char aa[100100];
int a[100100];
struct node{
int x,v;
};
bool cmp(node y,node yy){
return y.x>yy.x;
}
int main(){
int n,k;
node s[10000];
while(~scanf("%d %d",&n,&k))
{
scanf("%s",aa);
int ccc=0;
for(int i=0;i<n;i++){
a[i]=aa[i]-'0';
s[i].v=s[i].x=0;
if(a[i]==0)ccc++;
}
if(ccc==0){
printf("%d\n",n);
continue;
}
int p=0;
while(a[p]!=0){
p++;
}
int pp=0;
while(a[n-pp-1]!=0){
pp++;
}
if(k==0){
printf("%d\n",p);
continue;
}
int fp=0;
int flag=0;
for(int i=p;i<n-pp;i++){
if(a[i]!=fp){
fp=a[i];
flag++;
s[flag-1].v++;
s[flag].v++;
}
s[flag].x=s[flag].x+a[i];
}
sort(s,s+flag,cmp);
if(p==0&&pp==0){
int l=(k+1)/2;
int sum=0;
for(int i=0;i<flag&&i<l;i++){
sum=sum+s[i].x;
}
printf("%d\n",sum);
}
else if(p!=0&&pp==0){
int l=(k)/2;
int ll=(k+1)/2;
int sum=0,ans=0;
for(int i=0;i<flag&&i<l;i++){
sum=sum+s[i].x;
}
for(int i=0;i<flag&&i<ll;i++){
ans=ans+s[i].x;
}
printf("%d\n",max(sum+p,ans));
}
else if(p==0&&pp!=0){
int l=(k)/2;
int ll=(k+1)/2;
int sum=0,ans=0;
for(int i=0;i<flag&&i<l;i++){
sum=sum+s[i].x;
}
for(int i=0;i<flag&&i<ll;i++){
ans=ans+s[i].x;
}
printf("%d\n",max(sum+pp,ans));
}
else if(p!=0&&pp!=0){
int l=(k)/2;
int ll=(k+1)/2;
int lll=(k-1)/2;
int sum=0,ans=0,cnt=0;
for(int i=0;i<flag&&i<l;i++){
sum=sum+s[i].x;
}
for(int i=0;i<flag&&i<ll;i++){
ans=ans+s[i].x;
}
for(int i=0;i<flag&&i<lll;i++){
cnt=cnt+s[i].x;
}
printf("%d\n",max(max(sum+pp,ans),max(cnt+pp+p,sum+p)));
}
}
return 0;
}