BZOJ4566：[HAOI2016]找相同字符——题解

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+;
struct tree{
    int a[],fa,l;
}tr[N];
char s[N];
int tot,pos[N];
int a[N],w[N],len;
ll size[][N];
inline int insert(int p,int c,int id){
    int np=tr[p].a[c];
    if(np&&tr[np].l==tr[p].l+){
    size[id][np]++;
    return np;
    }
    np=++tot;tr[np].l=tr[p].l+;
    for(;p&&!tr[p].a[c];p=tr[p].fa)tr[p].a[c]=np;
    if(!p)tr[np].fa=;
    else{
    int q=tr[p].a[c];
    if(tr[p].l+==tr[q].l)tr[np].fa=q;
    else{
        int nq=++tot;tr[nq].l=tr[p].l+;
        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=;pos[]=;
    cin>>s+;
    int n=strlen(s+);
    for(int i=;i<=n;i++)pos[i]=insert(pos[i-],s[i]-'a',);
    cin>>s+;
    len=n;n=strlen(s+);len=max(len,n);
    for(int i=;i<=n;i++)pos[i]=insert(pos[i-],s[i]-'a',);
    for(int i=;i<=tot;i++)w[tr[i].l]++;
    for(int i=;i<=len;i++)w[i]+=w[i-];
    for(int i=;i<=tot;i++)a[w[tr[i].l]--]=i;
    ll ans=;
    for(int i=tot;i>;i--){
    size[][tr[a[i]].fa]+=size[][a[i]];
    size[][tr[a[i]].fa]+=size[][a[i]];
    ans+=size[][a[i]]*size[][a[i]]*(tr[a[i]].l-tr[tr[a[i]].fa].l);
    }
    printf("%lld\n",ans);
    return ;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++