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

【BZOJ3926】【Zjoi2015】诸神眷顾的幻想乡 广义后缀自动机

时间:2015-04-06 09:00:05      阅读:146      评论:0      收藏:0      [点我收藏+]

标签:bzoj3926   zjoi2015   诸神眷顾的幻想乡   广义后缀自动机   后缀自动机   

链接:

#include <stdio.h>
int main()
{
    puts("转载请注明出处[vmurder]谢谢");
    puts("网址:blog.csdn.net/vmurder/article/details/44891009");
}

题意、题解、数据、出题人标程

http://wjmzbmr.com/archives/zjoi-2015-day-1%e9%a2%98%e8%a7%a3/26

本蒟蒻的题解:

首先我们发现叶子节点很少,所以可以从每个叶子节点开始扫,然后就建立起了一棵大大的Trie树!然后统计有多少子串,就需要把Trie建立成广义(多串)后缀自动机。

然后怎么统计呢?
ans=i,isamdepi?deppai
为什么呢?
考虑abc建立起来的串,答案是6……嗯什么也发现不了。

然后abcb,我们发现第一个b字符和第二个b字符,它们的pa都是那个辅助节点。
发现是每个节点,从1开始到它的dep,都能构成长短不一的子串,但是其中或许会有重复,而对于这两个’b’,重复的部分都是pa的1~dep一部分,然后都减去了。
最终得到正确答案9。

代码:

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 101000
#define T 10
using namespace std;

struct Eli
{
    int v,next;
}e[N<<1];
int head[N],cnt,d[N];
inline void add(int u,int v)
{
    d[v]++;
    e[++cnt].v=v;
    e[cnt].next=head[u];
    head[u]=cnt;
}
int n,m,src[N];
struct SAM
{
    int son[N*20][T],pa[N*20],dep[N*20]; // 那个吧。这里开少了一倍TAT,竟然AC了
    int root,cnt,crs[N*20]; // 其实也不算奇怪,均摊下来差不多啦~(不是太好构造~)
    void begin(){crs[0]=root=cnt=1;}
    int newnode(int last)
    {
        dep[++cnt]=dep[last]+1;
        return cnt;
    }
    int add(int p,int x)
    {
        int np=son[p][x];
        if(np&&dep[np]==dep[p]+1)return np;
        else np=newnode(p); /* 这是与单串后缀自动机唯一有区别的部分*/

        while(p&&!son[p][x])son[p][x]=np,p=pa[p];
        if(!p)pa[np]=root;
        else {
            int q=son[p][x];
            if(dep[q]==dep[p]+1)pa[np]=q;
            else {
                int nq=newnode(p);
                pa[nq]=pa[q],pa[q]=pa[np]=nq;
                memcpy(son[nq],son[q],sizeof son[q]);
                while(p&&son[p][x]==q)son[p][x]=nq,p=pa[p];
            }
        }
        return np;
    }
    void dfs(int x,int p)
    {
        crs[x]=add(crs[p],src[x]);
        int i,v;
        for(i=head[x];i;i=e[i].next)
        {
            v=e[i].v;
            if(v==p)continue;
            dfs(v,x);
        }
    }
    bool vis[N*20];
    queue<int>q;
    long long bfs()
    {
        long long ret=0;
        q.push(root);
        int i,u,v;
        while(!q.empty())
        {
            u=q.front(),q.pop();
            if(u!=root)ret+=dep[u]-dep[pa[u]];
            for(i=0;i<m;i++)
            {
                v=son[u][i];
                if(vis[v])continue;
                vis[v]=true,q.push(v);
            }
        }
        return ret;
    }
}sam;
int main()
{
    freopen("test.in","r",stdin);

    int i,a,b;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)scanf("%d",&src[i]);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    sam.begin();
    for(i=1;i<=n;i++)if(d[i]==1)
        {sam.dfs(i,0);break;}
    cout<<sam.bfs()<<endl;
    return 0;
}

【BZOJ3926】【Zjoi2015】诸神眷顾的幻想乡 广义后缀自动机

标签:bzoj3926   zjoi2015   诸神眷顾的幻想乡   广义后缀自动机   后缀自动机   

原文地址:http://blog.csdn.net/vmurder/article/details/44891021

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