【HAOI2016】找相同字符

后缀自动机入门题

分别对两个串建后缀自动机，然后把endpos的大小算出来，最后在两个后缀自动机上一起dfs一遍并且算答案。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
const int maxn=5e5+10;
int n;
char s[maxn];
long long ans;
struct suda
{
	struct da{int ch[30],fa,len;}po[maxn];
	int las,tot,si[maxn],to[maxn],nt[maxn],st[maxn],topt;
	void init()
	{
		memset(po,0,sizeof po); memset(si,0,sizeof si);
		memset(to,0,sizeof to); memset(nt,0,sizeof nt);
		memset(st,0,sizeof st); las=tot=1; topt=0;
	}
	void insert(int x)
	{
		int p=las,np=las=++tot; po[np].len=po[p].len+1; si[np]=1;
		for (;p&&(!po[p].ch[x]);p=po[p].fa) po[p].ch[x]=np;
		if (!p) po[np].fa=1;
		else
		{
			int q=po[p].ch[x];
			if (po[q].len==po[p].len+1) po[np].fa=q;
			else
			{
				int nq=++tot; po[nq]=po[q];
				po[nq].len=po[p].len+1;
				po[np].fa=po[q].fa=nq;
				for (;p&&(po[p].ch[x]==q);p=po[p].fa) po[p].ch[x]=nq;
			}
		}
	}
	void add(int x,int y){to[++topt]=y; nt[topt]=st[x]; st[x]=topt;}
	void getsi(int x)
	{
		int p=st[x];
		while (p)
		{
			getsi(to[p]);
			si[x]+=si[to[p]];
			p=nt[p];
		}
	}
	void solve()
	{
		for (int i=2;i<=tot;++i) add(po[i].fa,i);
		getsi(1);
	}
}SAM[3];
void dfs(int x1,int x2)
{
	for (int i=0;i<26;++i)
	if (SAM[1].po[x1].ch[i] && SAM[2].po[x2].ch[i])
	{
/*		if (x1==1 && x2==1)
		{
			printf("%c %d %d\n",'a'+i,SAM[1].si[SAM[1].po[x1].ch[i]],SAM[2].si[SAM[2].po[x2].ch[i]]);
		}*/
		ans+=1ll*SAM[1].si[SAM[1].po[x1].ch[i]]*SAM[2].si[SAM[2].po[x2].ch[i]];
		dfs(SAM[1].po[x1].ch[i],SAM[2].po[x2].ch[i]);
	}
}
int main()
{
	for (int j=1;j<=2;++j)
	{
		scanf("%s",s+1); n=strlen(s+1); SAM[j].init();
		for (int i=1;i<=n;++i) SAM[j].insert(s[i]-'a');
		SAM[j].solve();
	}
	dfs(1,1); printf("%lld\n",ans);
return 0;
}