【bzoj3926】[Zjoi2015]诸神眷顾的幻想乡

最新推荐文章于 2019-05-07 21:30:00 发布

QinggegeQAQ

最新推荐文章于 2019-05-07 21:30:00 发布

阅读量1.2k

点赞数

分类专栏：后缀自动机

本文链接：https://blog.csdn.net/Qingegege/article/details/50767968

版权

后缀自动机专栏收录该内容

3 篇文章 0 订阅

订阅专栏

哼要先给那些根本没写清楚还发表的题解差评0.0如果不是clj的题解根本看不懂好嘛0.0
题意求一棵树上本质不同的子串的数量每个点出度<=20 叶子节点<=20。
从所有的叶子节点开始到根的每条路径的每一个前缀都对应一个子串（公共路径上会重复）
后缀自动机上可以识别全部的后缀也就是说按照从叶子节点到根的顺序插进去就相当于统计了所有的前缀于是就想到了后缀自动机~
后缀自动机不同的状态一定对应不同的子串并且整个自动机对应的一定是全部的子串所以ans+=len[i]-len[fa[i]]就是答案.
其实这个题用到的就是广义自动机就是对一棵tire树建立一个自动机
跟一般的后缀自动机的区别就是其实没什么区别就相当于从每一个叶子节点到根的路径上的点都从parent树根开始插入一遍这样重复的部分一定会在自动机中合并

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<map>
using namespace std;
const int N=400011;
int head[N],next[N],key[N],tot;
int a[N],du[N];
int last=1,cnt=1;
long long ans=0;
struct node
{
   int a[N*10][15],fa[N*10],len[N*10];
   int insert(int p,int x)
   {
     int np;
     np=last=++cnt;
     len[np]=len[p]+1;
     while(!a[p][x]&&p)a[p][x]=np,p=fa[p];
     if(!p)fa[np]=1;
     else
     {
       int q=a[p][x];
       if(len[p]+1==len[q])fa[np]=q;
       else
       {
         int nq=++cnt;len[nq]=len[p]+1;
         memcpy(a[nq],a[q],sizeof(a[q]));
         fa[nq]=fa[q];
         fa[np]=fa[q]=nq;
         while(a[p][x]==q)a[p][x]=nq,p=fa[p];
       }
     }
     return np;
   }
   void get()
   {
     for(int i=1;i<=cnt;i++)
     ans+=len[i]-len[fa[i]];
   }
}sam;

void add(int x,int y)
{
    tot++;
    next[tot]=head[x];
    head[x]=tot;
    key[tot]=y;
}
void dfs(int x,int fa,int p)
{
    int t=sam.insert(p,a[x]);
    for(int i=head[x];i;i=next[i])
     {
        int y=key[i];
        if(y!=fa)dfs(y,x,t);
     }
}
int main()
{
    int n,c;cin>>n>>c;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
        du[x]++,du[y]++;
    }
    for(int i=1;i<=n;i++)
    if(du[i]==1)dfs(i,0,1);
    sam.get();
    printf("%lld\n",ans);
}