字符串算法（算法竞赛）--拍案叫绝的KMP算法及其扩展

1、B站视频链接：F03【模板】KMP 算法_哔哩哔哩_bilibili

题目链接：【模板】KMP - 洛谷

#include <bits/stdc++.h>
using namespace std;
const int N=1000010;
int m,n;
char s[N],p[N];
int ne[N];

int main(){
	cin>>s+1>>p+1;
	m=strlen(s+1),n=strlen(p+1);
	//计算next函数，用于匹配失败时回跳 
	ne[1]=0; 
	for(int i=2,j=0;i<=n;i++){//以主串s为主 
		while(j&&p[i]!=p[j+1])j=ne[j];//回跳
		if(p[i]==p[j+1])j++;//指向前缀的末尾 
		ne[i]=j;// 相等前后缀的最长长度即为j 
	}
	//KMP匹配
	for(int i=1,j=0;i<=m;i++){//以模式串p为主 
		while(j&&s[i]!=p[j+1])j=ne[j];//匹配失败则回跳 
		if(s[i]==p[j+1])j++;//匹配成功则继续扫描往后模式串p
		if(j==n)printf("%d\n",i-n+1);//匹配完成 
	} 
	for(int i=1;i<=n;i++){
		printf("%d ",ne[i]);
	}
	return 0;
}

2、B站视频链接：F04 扩展 KMP(Z 函数)_哔哩哔哩_bilibili 

题目链接：【模板】扩展 KMP/exKMP（Z 函数） - 洛谷

#include <bits/stdc++.h>
using namespace std;
const int N=2e7+5;
char t[N],s[N];    
int z[N],p[N];

void get_z(char*s,int n){
  z[1]=n;
  for(int i=2,l,r=0;i<=n;i++){
    if(i<=r)z[i]=min(z[i-l+1],r-i+1);
    while(s[1+z[i]]==s[i+z[i]])z[i]++;
    if(i+z[i]-1>r)l=i,r=i+z[i]-1;
    // printf("i=%d z=%d [%d %d]\n",i,z[i],l,r);
  }
}
void get_p(char*s,int n,char*t,int m){
  for(int i=1,l,r=0;i<=m;i++){
    if(i<=r)p[i]=min(z[i-l+1],r-i+1);
    while(1+p[i]<=n&&i+p[i]<=m
      &&s[1+p[i]]==t[i+p[i]])p[i]++;
    if(i+p[i]-1>r)l=i,r=i+p[i]-1;  
  }
}
int main(){
  scanf("%s%s",t+1,s+1);
  int m=strlen(t+1),n=strlen(s+1);
  get_z(s,n);
  get_p(s,n,t,m);

  long long ans1=0,ans2=0;
  for(int i=1; i<=n; i++)
    ans1^=1LL*i*(z[i]+1);
  for(int i=1; i<=m; i++)
    ans2^=1LL*i*(p[i]+1);
  cout<<ans1<<endl<<ans2;
  return 0;
}