BZOJ 3926: [Zjoi2015]诸神眷顾的幻想乡 [广义后缀自动机 Trie]

3926: [Zjoi2015]诸神眷顾的幻想乡

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1124  Solved: 660
[Submit][Status][Discuss]

Description

 幽香是全幻想乡里最受人欢迎的萌妹子，这天，是幽香的2600岁生日，无数幽香的粉丝到了幽香家门前的太阳花田上来为幽香庆祝生日。 

粉丝们非常热情，自发组织表演了一系列节目给幽香看。幽香当然也非常高兴啦。 

这时幽香发现了一件非常有趣的事情，太阳花田有n块空地。在过去，幽香为了方便，在这n块空地之间修建了n-1条边将它们连通起来。也就是说，这n块空地形成了一个树的结构。 

有n个粉丝们来到了太阳花田上。为了表达对幽香生日的祝贺，他们选择了c中颜色的衣服，每种颜色恰好可以用一个0到c-1之间的整数来表示。并且每个人都站在一个空地上，每个空地上也只有一个人。这样整个太阳花田就花花绿绿了。幽香看到了，感觉也非常开心。 

粉丝们策划的一个节目是这样的，选中两个粉丝A和B（A和B可以相同），然后A所在的空地到B所在的空地的路径上的粉丝依次跳起来（包括端点），幽香就能看到一个长度为A到B之间路径上的所有粉丝的数目（包括A和B）的颜色序列。一开始大家打算让人一两个粉丝（注意：A,B和B,A是不同的，他们形成的序列刚好相反，比如红绿蓝和蓝绿红）都来一次，但是有人指出这样可能会出现一些一模一样的颜色序列，会导致审美疲劳。 

于是他们想要问题，在这个树上，一共有多少可能的不同的颜色序列（子串）幽香可以看到呢？ 

太阳花田的结构比较特殊，只与一个空地相邻的空地数量不超过20个。 

Input

 第一行两个正整数n,c。表示空地数量和颜色数量。 

第二行有n个0到c-1之间，由空格隔开的整数，依次表示第i块空地上的粉丝的衣服颜色。（这里我们按照节点标号从小到大的顺序依次给出每块空地上粉丝的衣服颜色）。 

接下来n-1行，每行两个正整数u,v，表示有一条连接空地u和空地v的边。 

Output

 一行，输出一个整数，表示答案。 

HINT

对于所有数据，1<=n<=100000, 1<=c<=10。 

---

我去真醉了.....SAM直接复制上一题，然后上一题只有01，然后就悲剧的找了好长时间SAM哪里出问题了.......

 

本题的重点是对20个Trie建SAM，方法是dfs中保存last然后建就行了.......

好吧其实我想了好长时间这样为什么对，然后看到了abclzr神犇写的才想明白 http://www.cnblogs.com/abclzr/p/6288505.html

 

直接对Trie建SAM与原本一个串建SAM唯一的不同是last可能已经有--c-->q了，我们有两种选择来处理

如果t[q].val==t[last].val+1

第一种是直接不管，这样t[np].par=q，np和q可以看作一个点，不受影响

第二种是管，直接让last走到q，也没关系

如果t[q].val!=t[last].val+1

这时会新建节点nq，然后t[q].par=t[np].par=nq 注意np和nq的Right是一样的(因为本来last有--c-->这个转移啊，所有的r都可以r+1)，但没关系，依旧看作一个点

 

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int N=2e5+5;
typedef long long ll;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1; c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0'; c=getchar();}
    return x*f;
}
int n,c,s[N];
struct node{
    int ch[10],par,val;
}t[N*20];
int sz=1,root=1;
void extend(int c,int &last){
    int q=t[last].ch[c];if(q&&t[q].val==t[last].val+1) {last=q;return;}
    int p=last,np=++sz;
    t[np].val=t[p].val+1;
    for(;p&&!t[p].ch[c];p=t[p].par) t[p].ch[c]=np;
    if(!p) t[np].par=root;
    else{
        int q=t[p].ch[c];
        if(t[q].val==t[p].val+1) t[np].par=q;
        else{
            int nq=++sz;
            t[nq]=t[q];t[nq].val=t[p].val+1;
            t[q].par=t[np].par=nq;
            for(;p&&t[p].ch[c]==q;p=t[p].par) t[p].ch[c]=nq;
        }
    }
    last=np;
}

struct edge{
    int v,ne;
}e[N];
int cnt,h[N],de[N];
inline void ins(int u,int v){
    cnt++;
    e[cnt].v=v;e[cnt].ne=h[u];h[u]=cnt;
    cnt++;
    e[cnt].v=u;e[cnt].ne=h[v];h[v]=cnt;
    de[u]++;de[v]++;
}
void dfs(int u,int fa,int last){
    extend(s[u],last);
    for(int i=h[u];i;i=e[i].ne) if(e[i].v!=fa) dfs(e[i].v,u,last);
}
void solve(){
    for(int i=1;i<=n;i++) if(de[i]==1) dfs(i,0,root);
    ll ans=0;
    for(int i=2;i<=sz;i++) ans+=t[i].val-t[t[i].par].val;
    printf("%lld",ans);
}
int main(){
    freopen("in","r",stdin);
    n=read();c=read();
    for(int i=1;i<=n;i++) s[i]=read();
    for(int i=1;i<=n-1;i++) ins(read(),read());
    solve();
}