题目链接:https://vjudge.net/contest/350917#problem/B
思路:
因为要包含最多的字符串,所以在匹配到s2最后一个位置时利用kmp的pre数组找到前面的最大匹配位置就好了。
代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 5e5+10;
char s1[N],s2[N],s3[N];
int a[5] = {0},pre[N] = {0};
int main(void)
{
scanf("%s%s",s1+1,s2+1);
int l1 = strlen(s1+1),l2 = strlen(s2+1);
pre[1] = 0;
for(int i=1,j=0;i<=l2;i++)
{
while(j>0 && s2[j+1] != s2[i+1]) j = pre[j];
if(s2[i+1] == s2[j+1]) j++;
pre[i+1] = j;
}
for(int i=1;i<=l1;i++) a[ s1[i] - '0' ]++;
for(int i=1,j=1;i<=l1;i++,j++)
{
if(s2[j] == '0')
{
if(a[0]>0) printf("0"),a[0]--;
else break;
}
else{
if(a[1]>0) printf("1"),a[1]--;
else break;
}
if(j == l2) j = pre[l2];
}
while(a[1]>0) printf("1"),a[1]--;
while(a[0]>0) printf("0"),a[0]--;
return 0;
}