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

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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+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/+

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

以上是关于BZOJ4566:[HAOI2016]找相同字符——题解的主要内容,如果未能解决你的问题,请参考以下文章

bzoj 4566 [Haoi2016]找相同字符SA

bzoj4566: [Haoi2016]找相同字符

●BZOJ 4566 [Haoi2016]找相同字符

Bzoj4566 [Haoi2016]找相同字符

Bzoj4566 [Haoi2016]找相同字符

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

(c)2006-2024 SYSTEM All Rights Reserved IT常识