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]);
}