HDU5129

HDU5129 - Yong Zheng's Death

---

做法：看题之后，就想到要建好trie树，然后输出非根节点数的平方，这样显然错了。因为会出现重算的情况，即：你有 ab,cde ,abc,de, abcd, e,这几个串那么 [ab, cde] [abc,de] [abcd,e] 都可以组成 abcde，那么就算了3次，答案就要减2。我们可以发现对于一个拼接好的串，他可能是由若干个对串构成的，这些串的分界线，将整个串分成了几份，我们要减去的就是多的份数，那么就考虑对于一个串存在多少份这样的串。对于一个节点S，他的fail指针指向的就是他的后缀对应的最长公共前缀，而S去除fail[S]的串，如果是其他前缀串的后缀，那么这个串会被重算。所以先在的问题就是对于每个串求出，以他为后缀的前缀的串的数目。而fail树对应节点的子树大小就是以这个串为后缀的前缀串的数目。参考了这篇BLOG

#include <bits/stdc++.h>
#define pb push_back
typedef long long ll;
const int N = 3e6+7;
using namespace std;
int n, T[N][26], cc, fail[N], fa[N], sum[N];
char str[66];
vector< int > G[N];
void init() {
    for(int i = 0; i <= cc; ++i) {
        memset(T[i],0,sizeof(T[i]));
        fail[i] = fa[i] = 0;
        G[i].clear();
    }
    cc = 0;
}
void ins(char str[]) {
    int len = strlen(str), now = 0;
    for(int i = 0; i < len ; ++i) {
        int t = str[i] - 'a';
        if(!T[now][t]) T[now][t] = ++cc, fa[cc] = now;
        now = T[now][t];
    }
}
void build_ac() {
    queue< int > q; while(!q.empty()) q.pop();
    for(int i = 0; i < 26; ++i) if(T[0][i]) q.push(T[0][i]);
    while( !q.empty() ) {
        int u = q.front(); q.pop();
        for(int i = 0; i < 26; ++i) {
            if(T[u][i]) fail[T[u][i]] = T[fail[u]][i], q.push(T[u][i]);
            else T[u][i] = T[fail[u]][i];
        }
    }
}
void dfs(int u) {
    sum[u] = 1;
    for(int i = 0; i < G[u].size(); ++i) {
        int v = G[u][i];
        dfs(v); sum[u] += sum[v];
    }
}
void getsum() {
    for(int i = 1; i <= cc; ++i) G[fail[i]].push_back(i);
    dfs(0);
}
ll cal() {
    ll ans = 1LL*cc*cc;
    for(int i = 1; i <= cc; ++i) if(fail[i]) {
        int p = i, q = fail[i];
        while(q) q=fa[q], p = fa[p];
        ans -= (sum[p]-1LL);
    }
    return ans;
}
int main() {
    while(scanf("%d",&n), n) {
        for(int i = 1; i <= n; ++i) scanf(" %s", str), ins(str);
        build_ac(); getsum();
        printf("%lld
", cal()); init();
    }
    return 0;
}