[JSOI2013]快乐的 JYY,洛谷P5685,广义SAM+PAM-优快云博客

正题

本质上与秩序魔咒毫无区别,就是换个方法来维护,原来只要看看是否都存在,现在要存在两边出现的次数.

#include<bits/stdc++.h>
using namespace std;

const int N=200010;
char s[N],t[N];
int n,m;
long long ans=0;

struct Suffix_AutoMaton{
	int tot,las;
	int len[N],ch[N][26],sum[N],fa[N][20],a[N],end[N],op[N][2];
	void insert(int c){
		if(ch[las][c]){
			int x=ch[las][c],p=las;
			if(len[x]==len[p]+1) las=x;
			else{
				int q=++tot;len[q]=len[p]+1;
				fa[q][0]=fa[x][0];fa[x][0]=q;
				for(int i=0;i<26;i++) ch[q][i]=ch[x][i];
				while(p!=-1 && ch[p][c]==x) ch[p][c]=q,p=fa[p][0];
				las=q;
			}
			return ;
		}
		int x=++tot,p=las;las=x;len[x]=len[p]+1;
		while(p!=-1 && !ch[p][c]) ch[p][c]=x,p=fa[p][0];
		if(p==-1) return ;
		if(len[ch[p][c]]==len[p]+1) fa[x][0]=ch[p][c];
		else{
			int q=++tot,tmp=ch[p][c];len[q]=len[p]+1;
			fa[q][0]=fa[tmp][0];fa[tmp][0]=fa[x][0]=q;
			for(int i=0;i<26;i++) ch[q][i]=ch[tmp][i];
			while(p!=-1 && ch[p][c]==tmp) ch[p][c]=q,p=fa[p][0];
		}
		return ;
	}
	void build(){
		scanf("%s %s",s+1,t+1);
		n=strlen(s+1);m=strlen(t+1);
		fa[0][0]=-1;for(int i=1;i<=n;i++) insert(s[i]-'A'),op[las][0]++,end[i]=las;
		las=0;for(int i=1;i<=m;i++) insert(t[i]-'A'),op[las][1]++;
		for(int i=1;i<=tot;i++) sum[len[i]]++;
		for(int i=1;i<=max(n,m);i++) sum[i]+=sum[i-1];
		for(int i=1;i<=tot;i++) a[sum[len[i]]--]=i;
		for(int i=tot;i>=1;i--) op[fa[a[i]][0]][0]+=op[a[i]][0],op[fa[a[i]][0]][1]+=op[a[i]][1];
		for(int i=1;i<=tot;i++){
			for(int k=1;k<20;k++) fa[a[i]][k]=fa[fa[a[i]][k-1]][k-1];
		}
	}
	void solve(int r,int lenth){
		int now=end[r];
		for(int k=19;k>=0;k--) if(fa[now][k]!=-1 && len[fa[now][k]]>=lenth) now=fa[now][k];
		ans+=1ll*op[now][0]*op[now][1];
	}
}SAM;

struct Palindrome_AutoMaton{
	int tot,las;
	int fail[N],ch[N][26],len[N];
	void init(){tot=1;fail[1]=0;fail[0]=1;len[1]=-1;}
	int gfail(int x,int pos){
		while(s[pos-len[x]-1]!=s[pos]) x=fail[x];
		return x;
	}
	void insert(int pos,int c){
		int p=gfail(las,pos);
		if(!ch[p][c]){
			int now=++tot;
			fail[now]=ch[gfail(fail[p],pos)][c];
			len[now]=len[p]+2;ch[p][c]=now;
			SAM.solve(pos,len[now]);
		}
		las=ch[p][c];
	}
	void build(){
		init();for(int i=1;i<=n;i++) insert(i,s[i]-'A');
		printf("%lld\n",ans);
	}
}PAM;

int main(){
	SAM.build();PAM.build();
}