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[N][10], fail[N], a[N];
	int last, root, tot;
	int p, q, np, nq;
	int c[N], rk[N], cnt[N], sum[N];

	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();
	}
}