【洛谷 P3346】 [ZJOI2015]诸神眷顾的幻想乡（后缀自动机）

题目链接

广义sam+不同子串个数。。

找到所有入度为\(0\)的点开始\(dfs\)，建出广义sam。

然后就是不同子串个数了

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN = 1000010;
struct SAM{
    int ch[12];
    int len, fa;
}sam[MAXN << 1];
int las = 1, cnt = 1, w[MAXN], in[MAXN];
long long f[MAXN << 1];
int len[MAXN], n, m, a, b;
struct Edge{
    int next, to;
}e[MAXN << 1];
int head[MAXN], num;
inline void Add(int from, int to){
    e[++num].to = to; e[num].next = head[from]; head[from] = num;
}
inline int add(int p, int c){
    int np = las = ++cnt;
    sam[np].len = sam[p].len + 1;
    for(; p && !sam[p].ch[c]; p = sam[p].fa) sam[p].ch[c] = np;
    if(!p) sam[np].fa = 1;
    else{
        int q = sam[p].ch[c];
        if(sam[q].len == sam[p].len + 1) sam[np].fa = q;
        else{
            int nq = ++cnt; sam[nq] = sam[q];
            sam[nq].len = sam[p].len + 1;
            sam[q].fa = sam[np].fa = nq;
            for(; p && sam[p].ch[c] == q; p = sam[p].fa) sam[p].ch[c] = nq;
        }
    }
    return np;
}
void dfs(int u){
    f[u] = 1;
    for(int i = 0; i < m; ++i)
        if(sam[u].ch[i]){
            if(!f[sam[u].ch[i]])
                dfs(sam[u].ch[i]);
            f[u] += f[sam[u].ch[i]];
        }
}
void Dfs(int u, int fa, int p){
    p = add(p, w[u]);
    for(int i = head[u]; i; i = e[i].next)
        if(e[i].to != fa)
            Dfs(e[i].to, u, p);
}
int main(){
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i)
        scanf("%d", &w[i]);
    for(int i = 1; i < n; ++i){
        scanf("%d%d", &a, &b);
        Add(a, b); Add(b, a);
        ++in[a]; ++in[b];
    }
    for(int i = 1; i <= n; ++i)
        if(in[i] == 1){
            las = 1;
            Dfs(i, 0, las);
        }
    dfs(1);
    printf("%lld\n", f[1] - 1);
    return 0;
}