KMP+Tire树(模板)

\(\color{Red}{KMP板子}\)

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+9;
int la,lb,j,kmp[maxn];
char a[maxn],b[maxn];
int main()
{
	cin>>a+1>>b+1;
	la=strlen(a+1),lb=strlen(b+1);
	for(int i=2;i<=lb;i++)//第一为默认为0
	{
		while(j&&b[i]!=b[j+1])	j=kmp[j];
		//ABCA的相同前后缀是A，如果想再增加
		//必须再添加一个B,也就是b[j+1]位置要等于b[i]
		if(b[j+1]==b[i])	j++;
		kmp[i]=j;
	}
	j=0;
	for(int i=1;i<=la;i++)
	{
		while(j&&b[j+1]!=a[i])	j=kmp[j];
		if(b[j+1]==a[i])	j++;
		if(j==lb)
		{
			cout<<i-lb+1<<endl;
			j=kmp[j];//找到后开始找下一个
		}
	}
	for(int i=1;i<=lb;i++)	cout<<kmp[i]<<" ";
}

\(\color{Red}{Tire树板子}\)

struct node{
	int zi[27];
}tire[maxn];int isok[maxn],tot;
void insert(string s)
{
	int now=0,len=s.length();
	for(int i=0;i<len;i++)
	{
		int ch=s[i]-'a'+1;
		if(!tire[now].zi[ch])
			tire[now].zi[ch]=++tot;
		now=tire[now].zi[ch];
	}
	isok[now]=1;//此节点是一个字串
}
bool ask(string s)
{
	int now=0,len=s.length();
	for(int i=0;i<len;i++)
	{
		int ch=s[i]-'a'+1;
		if(!tire[now].zi[ch])	return false;
		now=tire[now].zi[ch];
	}
	return true;
}