[BZOJ4566][HAOI2016]找相同子串

COGS传送门

用SAM重新写了一遍..

我的方法比较笨，先把两个串连在一起，算出来相同子串个数，同理算出s1和s2的子串个数。作差即可。

至于如何统计子串个数，首先toposort后搞出right集的大小，然后$C_{|right|}^{2} \times (step[i]-step[fa])$就是答案。

//BZOJ 4565
//by Cydiater
//2017.1.21
#include <iostream>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>
#include <map>
#include <ctime>
#include <iomanip>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <bitset>
#include <set>
#include <vector>
using namespace std;
#define ll long long
#define up(i,j,n)	for(int i=j;i<=n;i++)
#define down(i,j,n)	for(int i=j;i>=n;i--)
#define cmax(a,b)	a=max(a,b)
#define cmin(a,b)	a=min(a,b)
#define FILE		"find_2016"
const int MAXN=1e6+5;
const int oo=0x3f3f3f3f;
char s1[MAXN],s2[MAXN],s[MAXN];
int len1,len2;
struct SAM{
	int now,cnt,son[MAXN][30],step[MAXN],pre[MAXN],rnk[MAXN],label[MAXN];
	ll g[MAXN],ans;
	SAM(){now=1;cnt=1;}
	void Extend(int nxt){
		int p=now,np=++cnt;now=np;g[np]=1;
		step[np]=step[p]+1;
		for(;p&&!son[p][nxt];p=pre[p])son[p][nxt]=np;
		if(!p)pre[np]=1;
		else{
			int q=son[p][nxt],nq;
			if(step[q]==step[p]+1)pre[np]=q;
			else{
				step[(nq=++cnt)]=step[p]+1;
				memcpy(son[nq],son[q],sizeof(son[q]));
				pre[nq]=pre[q];
				pre[np]=pre[q]=nq;
				for(;son[p][nxt]==q;p=pre[p])son[p][nxt]=nq;
			}
		}
	}
	void Build(int len){
		up(i,1,len)Extend(s[i]-'a');
	}
	void Clear(){
		now=cnt=1;
		memset(pre,0,sizeof(pre));
		memset(son,0,sizeof(son));
		memset(step,0,sizeof(step));
		memset(g,0,sizeof(g));
		memset(label,0,sizeof(label));
		memset(rnk,0,sizeof(rnk));
	}
	ll col(int len){
		//topsort
		ans=0;
		up(i,1,cnt)label[step[i]]++;
		up(i,1,len)label[i]+=label[i-1];
		up(i,1,cnt)rnk[label[step[i]]--]=i;
		down(i,cnt,2){
			int node=rnk[i];
			if(g[node]>=2)
				ans+=(g[node]*(g[node]-1)>>1)*(ll)(step[node]-step[pre[node]]);
			g[pre[node]]+=g[node];
		}
		return ans;
	}
}sam;
namespace solution{
	void Prepare(){
		scanf("%s",s1+1);
		len1=strlen(s1+1);
		scanf("%s",s2+1);
		len2=strlen(s2+1);
	}
	void Solve(){
		ll ans1,ans2,ans3;
		memcpy(s,s1,sizeof(s));
		sam.Build(len1);
		ans1=sam.col(len1);
		sam.Clear();
		memcpy(s,s2,sizeof(s));
		sam.Build(len2);
		ans2=sam.col(len2);
		sam.Clear();
 
		memcpy(s,s1,sizeof(s));
		s[len1+1]='a'+26;
		up(i,len1+2,len1+len2+1)s[i]=s2[i-len1-1];
		sam.Build(len1+len2+1);
		ans3=sam.col(len1+len2+1);
		//cout<<ans1<<' '<<ans2<<' '<<ans3<<endl;
		ans3-=ans1+ans2;
		cout<<ans3<<endl;
	}
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	using namespace solution;
	Prepare();
	Solve();
	return 0;
}

看了看别人用SAM实现的代码，发现其实不必要那样。对于s1建一个sam，统计每个点的路径对应多少个子串。然后s2串在SAM上跑，经过一个点就统计上父亲点的sum（因为变换了位置）再加上当前对应的字符串的数量就行了。