https://www.lydsy.com/JudgeOnline/problem.php?id=4566
https://www.luogu.org/problemnew/show/P3181
给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数。两个方案不同当且仅当这两个子串中有一个位置不同。
广义后缀自动机,两个串各处理他们的size(或right?),然后对结点l排序,对于每个结点他们的size相乘即为答案。
……等等怎么WA了啊。
比如:
aba
abaa
这组数据ans=10。上面的想法是8
每个结点的贡献并不一定是这么算,该节点往左移(直到它的fa为止)的一段也可以匹配。
#include<cstdio> #include<iostream> #include<queue> #include<cstring> #include<algorithm> #include<cctype> using namespace std; typedef long long ll; const int N=8e5+5; struct tree{ int a[26],fa,l; }tr[N]; char s[N]; int tot,pos[N]; int a[N],w[N],len; ll size[2][N]; inline int insert(int p,int c,int id){ int np=tr[p].a[c]; if(np&&tr[np].l==tr[p].l+1){ size[id][np]++; return np; } np=++tot;tr[np].l=tr[p].l+1; for(;p&&!tr[p].a[c];p=tr[p].fa)tr[p].a[c]=np; if(!p)tr[np].fa=1; else{ int q=tr[p].a[c]; if(tr[p].l+1==tr[q].l)tr[np].fa=q; else{ int nq=++tot;tr[nq].l=tr[p].l+1; memcpy(tr[nq].a,tr[q].a,sizeof(tr[q].a)); tr[nq].fa=tr[q].fa;tr[q].fa=tr[np].fa=nq; for(;p&&tr[p].a[c]==q;p=tr[p].fa)tr[p].a[c]=nq; } } size[id][np]++; return np; } int main(){ tot=1;pos[0]=1; cin>>s+1; int n=strlen(s+1); for(int i=1;i<=n;i++)pos[i]=insert(pos[i-1],s[i]-‘a‘,0); cin>>s+1; len=n;n=strlen(s+1);len=max(len,n); for(int i=1;i<=n;i++)pos[i]=insert(pos[i-1],s[i]-‘a‘,1); for(int i=1;i<=tot;i++)w[tr[i].l]++; for(int i=1;i<=len;i++)w[i]+=w[i-1]; for(int i=1;i<=tot;i++)a[w[tr[i].l]--]=i; ll ans=0; for(int i=tot;i>1;i--){ size[0][tr[a[i]].fa]+=size[0][a[i]]; size[1][tr[a[i]].fa]+=size[1][a[i]]; ans+=size[0][a[i]]*size[1][a[i]]*(tr[a[i]].l-tr[tr[a[i]].fa].l); } printf("%lld\n",ans); return 0; }
+++++++++++++++++++++++++++++++++++++++++++
+本文作者:luyouqi233。 +
+欢迎访问我的博客:http://www.cnblogs.com/luyouqi233/+
+++++++++++++++++++++++++++++++++++++++++++
原文地址:https://www.cnblogs.com/luyouqi233/p/8799329.html
时间: 2024-11-06 03:37:23