CF1138D.Camp Schedule

传送门

虽然是D，但是还是Sb题，把模式串跑一遍KMP，然后把按顺序放，每次放完之后跳到对应的前缀，继续放。
如果最后1的数量还有剩，再将最后的位数全部放1
代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=5e5+10;
char p[maxn],s[maxn];
int num,num1,len,len1,nxt[maxn],ans;
int main()
{
    scanf("%s",p+1),scanf("%s",s+1);
    len=strlen(p+1),len1=strlen(s+1);
    for(rg int i=1;i<=len;i++)if(p[i]=='1')num++;
    for(rg int i=1;i<=len1;i++)if(s[i]=='1')num1++;
    if(num<num1||len<len1)
    {
        for(rg int i=1;i<=len;i++)printf("%c",p[i]);
        printf("\n");return 0;
    }
    for(rg int i=2,j=0;i<=len1;i++)
    {
        while(j&&s[j+1]!=s[i])j=nxt[j];
        if(s[j+1]==s[i])j++;
        nxt[i]=j;
    }
    ans=len+1;
    for(rg int i=1,j=0;i<=len;i++)
    {
        if(num&&s[j+1]=='1')num--,p[i]='1',j++;
        else if(s[j+1]=='0')j++,p[i]='0';
        else {ans=i;break;}
        if(j==len1)j=nxt[j];
    }
    for(rg int i=ans;i<=len;i++)p[i]='0';
    while(num)
    {
        ans--;
        if(p[ans]=='1')num++;
        num--,p[ans]='1';
    }
    for(rg int i=1;i<=len;i++)printf("%c",p[i]);
}

转载于:https://www.cnblogs.com/lcxer/p/10499893.html