p5410——KMP扩展模板题

题目链接

#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;
}