BZOJ4566: [Haoi2016]找相同字符

Description

给定两个字符串，求出在两个字符串中各取出一个子串使得这两个子串相同的方案数。两个方案不同当且仅当这两

个子串中有一个位置不同。

Input

两行，两个字符串s1，s2，长度分别为n1，n2。1 <=n1, n2<= 200000，字符串中只有小写字母

Output

输出一个整数表示答案

Sample Input

aabb
bbaa

Sample Output

10

 

裸题，用SAM来做的话先建出第一个串的后缀自动机，把第二个串拿上去运行，不完整节点直接计算，完整节点打个懒标记最后扫一遍就行了。

#include<cstdio>
#include<cctype>
#include<queue>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
typedef long long ll;
const int maxn=400010;
int to[maxn][26],l[maxn],fa[maxn],last=1,cnt=1;
int f[maxn],siz[maxn],od[maxn],c[maxn];
void extend(int c) {
	int p,q,np,nq;p=last;l[last=np=++cnt]=l[p]+1;siz[np]=1;
	for(;!to[p][c];p=fa[p]) to[p][c]=np;
	if(!p) fa[np]=1;
	else {
		q=to[p][c];
		if(l[p]+1==l[q]) fa[np]=q;
		else {
			l[nq=++cnt]=l[p]+1;
			fa[nq]=fa[q];fa[q]=fa[np]=nq;
			memcpy(to[nq],to[q],sizeof(to[nq]));
			for(;to[p][c]==q;p=fa[p]) to[p][c]=nq;
		}
	}
}
char s[maxn];
int main() {
	scanf("%s",s);
	for(int i=0;s[i];i++) extend(s[i]-'a');
	scanf("%s",s);
	int p=1,len=0;ll ans=0;
	rep(i,1,cnt) c[l[i]]++;
	rep(i,1,cnt) c[i]+=c[i-1];
	dwn(i,cnt,1) od[c[l[i]]--]=i;
	dwn(i,cnt,1) siz[fa[od[i]]]+=siz[od[i]];
	for(int i=0;s[i];i++) {
		int c=s[i]-'a';
		if(to[p][c]) len++,p=to[p][c];
		else {
			while(p&&!to[p][c]) p=fa[p];
			if(p) len=l[p]+1,p=to[p][c];
			else p=1,len=0;
		}
		ans+=(ll)siz[p]*(len-l[fa[p]]);f[fa[p]]++;
	}
	dwn(i,cnt,1) {
		int x=od[i];f[fa[x]]+=f[x];
		ans+=(ll)f[x]*siz[x]*(l[x]-l[fa[x]]);
	}
	printf("%lld\n",ans);
	return 0;
}