码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ 3238 [Ahoi2013]差异

时间:2016-06-16 17:45:25      阅读:181      评论:0      收藏:0      [点我收藏+]

标签:

AC通道:http://www.lydsy.com/JudgeOnline/problem.php?id=3238

技术分享

发现前面的len(Ti)+len(Tj)都是定值,所以其实我们的目标就是求出lcp(Ti,Tj)来.

lcp说实话就是一段子串对吧,但是不太好找.所以我们可以把串反过来.

这样的话,原图中后缀的最长公共前缀就变成了新串中前缀的最长公共后缀.

公共后缀是十分好求的,我们熟知parent边就是一个不断取后缀的过程,然后两者不断删去首字母,直到后缀相同的时候也就到了它们在parent树上的LCA的最长长度.

在一棵parent树上,考虑LCA可以是多少个点对的LCA.[应该是一个基本树型DP]

因为我们求的是前缀的公共后缀,所以每个状态只能取mx作为使用者.那么其实只要知道LCA的所有儿子的right集合大小即可.

right集合的每个位置分别表示了一段前缀[其实也就是这个节点下面的子树大小],然后与它父亲的其它可能的前缀的可能相乘 就得到了有多少个点对.

 

然后怎么求right集合的大小呢?

有一个性质就是parent树是一个right集合合并的过程,那么一个点的right集合大小就是其所有子树right集合大小之和.

还记得之前有画过一个图,现在再扯过来说一下:

技术分享

那么我们要是能把叶子节点的值赋值为1,然后沿着parent往上走就可以得到了.

叶子节点怎么找呢?其实很简单...当你每次extend的时候,你会增加一个新的节点np,这个节点中就只包含一个元素{n},然后其它所有right集合中有n的都是它的祖先,所以它就是一个叶子节点,需要+1.

而有时候我们将一个节点拆分成两个的时候,并没有产生新的元素,所以不是叶子节点不需要+1.

当然有时候你不确定的原因是有的,比如:"abab"中的"ab"它明明有儿子,但是它确实又是一个叶子节点,提供了它自己right集合里面的2这个元素.

("ab")={2,4}--b-->("bab")={4}

     --无-->("无ab")={2}

所以每个前缀都一定在叶子节点上...好神奇啊...然后大家记住就好了.

 

当然还有一个拓扑序的完成,这个就是按照长度来弄一下顺序就好了,越长的当然越在后面.

看代码就好.

技术分享
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

typedef long long ll;
const int maxn=1000010;

int n,last=1,cnt=1;
int a[maxn][26],mx[maxn],fa[maxn];
int T[maxn],Seq[maxn];
int f[maxn],w[maxn];
char ch[maxn];

void extend(int c){
    int p=last,np=last=++cnt;
    mx[np]=mx[p]+1; f[np]=w[np]=1;
    while(!a[p][c] && p) a[p][c]=np,p=fa[p];
    if(!p) fa[np]=1;
    else{
        int q=a[p][c];
        if(mx[q]==mx[p]+1) fa[np]=q;
        else{
            int nq=++cnt;mx[nq]=mx[p]+1;
            memcpy(a[nq],a[q],sizeof(a[q]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            while(a[p][c]==q) a[p][c]=nq,p=fa[p];
        }
    }
}

void get_order(){
    for(int i=1;i<=cnt;i++) T[mx[i]]++;
    for(int i=1;i<=n;i++) T[i]+=T[i-1];
    for(int i=1;i<=cnt;i++) Seq[T[mx[i]]--]=i;
}

ll solve(){
    ll ans=0;
    for(int i=cnt;i>=1;i--) f[fa[Seq[i]]]+=f[Seq[i]];
    for(int i=cnt;i>=1;i--){
        int x=Seq[i];
        ans+=(ll)f[x]*w[fa[x]]*mx[fa[x]];
        w[fa[x]]+=f[x];
    }
    return ans;
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("3238.in","r",stdin);
    freopen("3238.out","w",stdout);
#endif

    scanf("%s",ch);
    n=strlen(ch);
    for(int i=n-1;i>=0;i--) extend(ch[i]-a);
    
    ll ans=(ll)(1+n)*n*(n-1)/2;
    
    get_order();
    ans-=solve()*2;
    
    printf("%lld",ans);
    return 0;
}
View Code

 

BZOJ 3238 [Ahoi2013]差异

标签:

原文地址:http://www.cnblogs.com/Robert-Yuan/p/5591627.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!