【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,i\in sam} dep_i-dep_{pa_i}$
为什么呢？
考虑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;
}