P3181 [HAOI2016]找相同字符

题目链接：[Haoi2016]找相同字符 - 题目 - 黑暗爆炸OJ (darkbzoj.tk)

考虑分别对两个字符串建SAM，在SAM上通过next转移，路径上形成的字符串必定相同。先预处理出每个状态对应所有子串的出现次数。然后通过dfs同时在两个SAM上转移，每个节点形成的贡献便是$siz[u]\ast siz[v]$

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define se second
#define fi first
#define LINF 0x3f3f3f3f3f3f3f3f
#define INF 0x3f3f3f3f
#define pii pair<int,int>
#define ll long long
const int mod=998244353;
const int N=800005;
struct node{
	map<int,int>nex[N];
	int link[N],len[N];
	long long siz[N];
	int idx=1;
	int last=1;
	ll ans=0;
	vector<int>vec[N];
	void sam_extend(int c){
		int cur=++idx;siz[cur]=1;
		len[cur]=len[last]+1;//增加一个新状态，长度为上一个+1
		int p=last;
		while(p&&!nex[p].count(c)){//向上遍历所有后缀链接，判断转移当前点
			nex[p][c]=cur;//如果没有c可以转移，那么p状态所有子串+c构成的子串加入到cur中
			p=link[p];
		}
		if(!p){//如果直到根都没有，直接link根
			link[cur]=1;
		}
		else{//遇到已经存在可以转移c的状态，说明p的所有子串+c构成的串的位置的集合!=cur的位置集合，那么考虑下面两种情况
			int q=nex[p][c];
			if(len[p]+1==len[q]){//如果q最长子串是由p+c构成的，那么cur可以直接link到q点，否则不满足link上所有子串是cur的子串这个条件
				link[cur]=q;
			}
			else{//如果不满足link条件，考虑将link分成两部分
				int clone=++idx;//克隆一份
				len[clone]=len[p]+1;//长度为p+1，保证cur能link上
				nex[clone]=nex[q];//子串一样能+c转移
				link[clone]=link[q];//原q的所有子串位置相同，那么link也应该相同
				while(p&&nex[p][c]==q){//将所有转移转到新的clone上
					nex[p][c]=clone;
					p=link[p];
				}
				link[cur]=link[q]=clone;
			}
		}
		last=cur;
	}
	void dfs(int u){
		for(auto x:vec[u]){
			dfs(x);
			siz[u]+=siz[x];
		}
	}
	void getcnt(){
		for(int i=1;i<=idx;i++){
			vec[link[i]].push_back(i);
		}
		dfs(1);
	}
}sa1,sa2;
ll ans=0;
void dfs(int u,int v){
	if(u!=1&&v!=1){
		ans+=1ll*sa1.siz[u]*sa2.siz[v];
	}
	for(int i=0;i<26;i++){
		if(sa1.nex[u].count(i)&&sa2.nex[v].count(i)){
			dfs(sa1.nex[u][i],sa2.nex[v][i]);
		}
	}
}
void solve(){
	string s1;cin>>s1;
	string s2;cin>>s2;
	for(auto x:s1){
		sa1.sam_extend(x-'a');
	}
	for(auto x:s2){
		sa2.sam_extend(x-'a');
	}
	sa1.getcnt();sa2.getcnt();
	dfs(1,1);
	cout<<ans;
}
int main(){
	solve();
	return 0;
}

但注意到，转移图是一个DAG图，复杂度会出问题才对，也就是说这道题的数据水了，不过可以通过记忆化dp优化时间。