HDU5129 - Yong Zheng's Death

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\n", cal()); init();}return 0;
}