#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
const int NN=20000005;
int m,n;
int extend[NN];
int nexts[NN];
char S[NN],T[NN];
//两段相似代码
//nexts[i]: T[i]...T[m - 1]与T的最长相同前缀长度;
//extend[i]: S[i]...S[n - 1]与T的最长相同前缀长度。
void getnext()
{
int a=0,p=0;
nexts[0]=m;
for(int i=1; i<m; i++)
{
if(i>=p||i+nexts[i-a]>=p)
{
if(i>=p)p=i;
while(p<m&&T[p]==T[p-i])p++;
nexts[i]=p-i;
a=i;
}
else nexts[i]=nexts[i-a];
}
}
void extend_KMP()
{
int a=0,p=0;
getnext();
for(int i=0; i<n; i++)
{
if(i>=p||i+nexts[i-a]>=p)
{
if(i>=p)p=i;
while(p<n&&p-i<m&&S[p]==T[p-i])p++;
extend[i]=p-i;
a=i;
}
else extend[i]=nexts[i-a];
}
}
int main()
{
scanf("%s%s",&S,&T);//用char不失为一种良策(不对string太执着)一直cin一直TLE
m=strlen(T);
n=strlen(S);
extend_KMP();
long long t1=0,t2=0;
for(int i=0; i<m; i++)
t1^=(long long)1*(i+1)*(nexts[i]+1);
for(int i=0; i<n; i++)
t2^=(long long)1*(i+1)*(extend[i]+1);
cout<<t1<<endl<<t2<<endl;
return 0;
}
p5410——KMP扩展模板题
最新推荐文章于 2024-08-26 21:46:59 发布