拓展KMP

模板题

/*
拓展KMP和z函数
一般z函数用来匹配自己
拓展KMP用来和别人进行匹配
用匹配串进行一次KMP匹配
然后与模式串进行匹配
*/
 
#include <bits/stdc++.h>
using namespace std;
const int M=2e7+5;
using ll=long long;
 
int z[M],p[M];
 
void Z(char s[]) {
    int n=strlen(s+1);
    z[1]=n;
    for(int i=2,l=0,r=0;i<=n;i++) {
        if(i<=r)z[i]=min(z[i-l+1],r-i+1);//最大的长度
        while(i+z[i]<=n&&s[1+z[i]]==s[i+z[i]])z[i]++;//暴力匹配
        if(i+z[i]-1>r)l=i,r=i+z[i]-1;
    }
}
 
void exkmp(char *s,char *t) {
    Z(t);
    int n=strlen(s+1),m=strlen(t+1);
    for(int i=1,l=0,r=0;i<=n;i++) {
        if(i<=r)p[i]=min(z[i-l+1],r-i+1);
        while(i+p[i]<=n&&1+p[i]<=m&&s[i+p[i]]==t[1+p[i]])p[i]++;
        if(i+p[i]-1>r)l=i,r=i+p[i]-1;
    }
}
 
char a[M],b[M];
 
int main() {
    scanf("%s",a+1);
    scanf("%s",b+1);
    int n=strlen(a+1),m=strlen(b+1);
    exkmp(a,b);
    ll ans1=0,ans2=0;
    for(int i=1;i<=m;i++)ans1^=1ll*i*(z[i]+1);
    for(int i=1;i<=n;i++)ans2^=1ll*i*(p[i]+1);
    cout<<ans1<<endl<<ans2<<endl;
    return 0;
}