HDU 4436 (后缀自动机)

HDU 4436 str2int

Problem : 给若干个数字串，询问这些串的所有本质不同的子串转换成数字之后的和。

Solution : 首先将所有串丢进一个后缀自动机。由于这道题询问的是不同的子串之间的计数，不涉及子串的数目，因此考虑用后缀链即后缀自动机的nt转移数组来做。首先将所有节点进行按照距离大小进行基数排序，然后从小到大枚举每个节点的出边，假设当前点为u，转移到v，那么更新的状态为cnt[v] += cnt[u], sum[v] += sum[u] + cnt[u] * ch，cnt[u]表示的是到达当前节点的子串的数量，sum[u]表示的是该节点所表示的串的答案。

需要注意的是具有前导0的串对答案没有贡献。

#include <string>
#include <iostream>

using namespace std;

const int N = 200008;
const int mo = 2012;

struct Suffix_Automaton
{
int nt
[10], fail
, a
;
int last, root, tot;
int p, q, np, nq;
int c
, rk
, cnt
, sum
;

int newnode(int len)
{
for (int i = 0; i < 10; ++i) nt[tot][i] = -1;
fail[tot] = -1; a[tot] = len;
c[tot] = rk[tot] = cnt[tot] = sum[tot] = 0;
return tot++;
}
void clear()
{
tot = 0;
last = root = newnode(0);
}
void insert(int ch)
{
p = last; last = np = newnode(a[p] + 1);
for (; ~p && nt[p][ch] == -1; p = fail[p]) nt[p][ch] = np;
if (p == -1) fail[np] = root;
else
{
q = nt[p][ch];
if (a[p] + 1 == a[q]) fail[np] = q;
else
{
nq = newnode(a[p] + 1);
for (int i = 0; i < 10; ++i) nt[nq][i] = nt[q][i];
fail[nq] = fail[q];
fail[q] = fail[np] = nq;
for (; ~p && nt[p][ch] == q; p = fail[p]) nt[p][ch] = nq;
}
}
}
void solve()
{
for (int i = 0; i < tot; ++i) c[a[i]]++;
for (int i = 1; i < tot; ++i) c[i] += c[i - 1];
for (int i = 0; i < tot; ++i) rk[--c[a[i]]] = i;
cnt[root] = 1;
for (int i = 0; i < tot; ++i)
{
int u = rk[i];
for (int j = 0; j < 10; ++j)
if (~nt[u][j])
{
if (i == 0 && j == 0) continue;
int v = nt[u][j];
cnt[v] += cnt[u ];
sum[v] += sum[u] * 10 + j * cnt[u];
sum[v] %= mo;
}
}
int ans = 0;
for (int i = 0; i < tot; ++i)
{
ans += sum[i];
ans %= mo;
}
cout << ans << endl;
}

}sam;

int main()
{
cin.sync_with_stdio(0);
int n;
while (cin >> n)
{
string s;
sam.clear();
for (int i = 1; i <= n; ++i)
{
cin >> s;
sam.last = 0;
for (int j = 0, len = s.length(); j < len; ++j)
sam.insert(s[j] - '0');
}
sam.solve();
}
}