链接:https://codeforces.com/contest/1137/problem/B
The new camp by widely-known over the country Spring Programming Camp is going to start soon. Hence, all the team of friendly curators and teachers started composing the camp's schedule. After some continuous discussion, they came up with a schedule ss, which can be represented as a binary string, in which the ii-th symbol is '1' if students will write the contest in the ii-th day and '0' if they will have a day off.
At the last moment Gleb said that the camp will be the most productive if it runs with the schedule tt (which can be described in the same format as schedule ss). Since the number of days in the current may be different from number of days in schedule tt, Gleb required that the camp's schedule must be altered so that the number of occurrences of tt in it as a substring is maximum possible. At the same time, the number of contest days and days off shouldn't change, only their order may change.
Could you rearrange the schedule in the best possible way?
Input
The first line contains string ss (1⩽|s|⩽5000001⩽|s|⩽500000), denoting the current project of the camp's schedule.
The second line contains string tt (1⩽|t|⩽5000001⩽|t|⩽500000), denoting the optimal schedule according to Gleb.
Strings ss and tt contain characters '0' and '1' only.
Output
In the only line print the schedule having the largest number of substrings equal to tt. Printed schedule should consist of characters '0' and '1' only and the number of zeros should be equal to the number of zeros in ss and the number of ones should be equal to the number of ones in ss.
In case there multiple optimal schedules, print any of them.
Examples
input
Copy
101101 110
output
Copy
110110
input
Copy
10010110 100011
output
Copy
01100011
input
Copy
10 11100
output
Copy
01
Note
In the first example there are two occurrences, one starting from first position and one starting from fourth position.
In the second example there is only one occurrence, which starts from third position. Note, that the answer is not unique. For example, if we move the first day (which is a day off) to the last position, the number of occurrences of tt wouldn't change.
In the third example it's impossible to make even a single occurrence.
代码:
#include<bits/stdc++.h>
using namespace std;
char s[10000001], t[10000001];
long long nxt[10000001],n,k,m;
void kmp(long long x, char*a)
{
int i,j;
for(nxt[0]=j=-1,i=1;i<x;nxt[i++]=j)
{
while(~j&&a[j+1]!=a[i])
j=nxt[j];
if(a[j+1]==a[i])
j++;
}
}
long long c[2];
int main()
{
cin>>s;
cin>>t;
n=strlen(t);
kmp(n,t);
k=nxt[n-1];
m=strlen(s);
for(int i=0;i<m;i++)
c[s[i]-'0']++;
for(int i=0,j=0;i<m;i++)
{
if(!c[0])
cout<<1;
else if(!c[1])
cout<<0;
else
{
cout<<t[j];
c[t[j]-'0']--;
if(j==n-1)
j=nxt[j];
j++;
}
}
}