CodeForces - 1137B （kmp）

题目链接：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;
}