codeforces1137B kmp(fail的妙用)

时间:2023-03-10 04:03:23
codeforces1137B    kmp(fail的妙用)

题目传送门

题意:给出$s$和$t$两个串,让你构造出一个答案串,使得答案串中的01数量和s一样,并且使$t$在答案串中作为子串出现次数最多。

思路:

  要想出现的次数尽可能多,那么就要重复的利用,哪一部分是可以重复利用的呢?就是前缀和后缀相同的部分,然后我们就想到了$kmp$算法中$fail$函数就是求这个东西的,那么我们先对t串fail一遍得到$next$数组,然后先使前缀出现一次,然后就使除了前缀以外的后缀尽可能出现的多,这样得到的答案串必定是最多的,最后把剩余的01输出就可以了。

  记住扫描字符串的for循环千万不要用(i<strlen(s))当条件,否则就是个$n2$的算法了(你会tle15)。

#include<bits/stdc++.h>
#define clr(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int maxn=;
int n,f[maxn],a,b;
char s[maxn],p[maxn];
void fail(){
f[]=-;
for(int j=;j<n;j++)
{
for(int i=f[j-];;i=f[i])
{
if(p[j]==p[i+]){
f[j]=i+;
break;
}else if(i==-)
{
f[j]=-;
break;
}
}
}
for(int i=;i<n;i++)f[i]++;
}
int main(){
while(scanf("%s",s)!=EOF)
{
scanf("%s",p);
n=strlen(p);
fail();
a=b=;
int si=strlen(s);
for(int i=;i<si;i++)
{
if(s[i]=='')a++;
else b++;
}
int xx=,yy=;
for(int i=;i<f[n-];i++)
{
if(p[i]=='')xx++;
else yy++; }
if(a>=xx&&b>=yy)
{
for(int i=;i<f[n-];i++)printf("%c",p[i]);
a-=xx,b-=yy;
int xxx=,yyy=;
for(int i=f[n-];i<n;i++)
{
if(p[i]=='')xxx++;
else yyy++;
}
int flag=;
while(a>&&b>)
{
if(a<xxx||b<yyy)break;
for(int i=f[n-];i<n;i++)
{
printf("%c",p[i]);
}
a-=xxx,b-=yyy;
}
while(a>){
printf("");
a--;
}
while(b>){
printf("");
b--;
}
puts("");
}else{
printf("%s\n",s);
}
}
}