Code froces 1137B Camp Schedule 【贪心+KMP】

传送门

题意：给你一个文本串a，和一个模式串b，你可以重组a，使a中出现b的次数最多，可以叠在一起（a：10101，b=101，b最多可以出现过2次）。

思路：两个串里面只有 0 和 1 字符，我们需要记录 a 串中的 0 1个数。然后去构造 b 串 ，如果已经构造完一个 b 串了，我们想要 b 串出现的次数最多，我们让已经构造出来的串的后半段（即b串的后半段）尽可能与 b 的前半段重叠部分最长，那不就是 b 串 next数组的定义吗？

附上代码：

///#include<bits/stdc++.h>
///#include<unordered_map>
///#include<unordered_set>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<bitset>
#include<set>
#include<stack>
#include<map>
#include<list>
#include<new>
#include<vector>
#define MT(a,b) memset(a,b,sizeof(a));
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const double E=2.718281828459;
const int INF=0x3f3f3f3f;

char a[500005];
char b[500005];
string ans="";
int nex[500005];
int zero=0,one=0;

int main()
{
    memset(nex,0,sizeof(nex));
    cin>>a+1>>b+1;
    int d1=strlen(a+1);
    int d2=strlen(b+1);
    if(d2>d1)
        cout<<a+1<<endl;
    else
    {
        nex[1]=0;
        int k=0;
        for(int i=2;i<=d2;i++)
        {
            while(k&&b[i]!=b[k+1])
                k=nex[k];
            if(b[i]==b[k+1])
                k++;
            nex[i]=k;
        }
        for(int i=1;i<=d1;i++)
            a[i]=='0'?zero++:one++;
        int sign=1;
        while(zero>0&&one>0)
        {
            if(b[sign]=='0')
            {
                if(zero>0)
                {
                    ans+='0';
                    zero--;
                }
                else
                    break;
            }
            else
            {
                if(one>0)
                {
                    ans+='1';
                    one--;
                }
                else
                    break;
            }
            sign++;
            if(sign==d2+1)///构造完一个b串以后
                sign=nex[d2]+1;
        }
        ///将剩下的0和1加在答案上面
        while(zero--)
            ans+='0';
        while(one--)
            ans+='1';
        cout<<ans<<endl;
    }
    return 0;
}