BZOJ3926 諸神眷顧的幻想鄉

Problem :

給一個n個節點的樹（n<=10^5), 每個點有一種顏色(c<=10), 詢問所有點對之間路徑組成字符串的種類。保證葉子節點小于等于20.

Solution :

分別從每個葉子節點開始遍歷整棵樹，將遍歷到的字符串加入后綴自動機。

只需要修改后綴自動機的last，就可以實現添加多串。

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 * 15 * 2;

vector <int> eg[N];

int cl[N];

int deg[N];

int n, c;

struct Suffix_Automanon

{

	int nt[N][15], fail[N], a[N];

	int p, q, np, nq;

	int tot, root;

	int newnode(int len)

	{

		for (int i = 0; i < 15; ++i) nt[tot][i] = -1;

		fail[tot] = -1; a[tot] = len;

		return tot++;

	}

	void clear()

	{

		tot = 0;

		root = newnode(0);

	}

	void insert(int ch, int &last)

	{

		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[q] == a[p] + 1) fail[np] = q;

			else

			{

				nq = newnode(a[p] + 1);

				for (int i = 0; i < 15; ++i) nt[nq][i] = nt[q][i];

				fail[nq] = fail[q]; fail[q] = fail[np] = nq;

				for (; ~q && nt[p][ch] == q; p = fail[p]) nt[p][ch] = nq;

			}

		}

	}

	void solve()

	{

		long long ans = 0;

		for (int i = 1; i < tot; ++i)

			ans += a[i] - a[fail[i]];

		cout << ans << "\n";

	}

}sam;

void dfs(int u, int fa, int now)

{

	sam.insert(cl[u], now);

	for (int i = 0; i < (int)eg[u].size(); ++i)

	{

		int v = eg[u][i];

		if (v != fa) dfs(v, u, now);

	}

}

int main()

{

	cin.sync_with_stdio(0);

	cin >> n >> c;

	for (int i = 1; i <= n; ++i) cin >> cl[i];

	for (int i = 1; i <= n; ++i) eg[i].clear(), deg[i] = 0;

	for (int i = 1; i <  n; ++i)

	{

		int u, v;

		cin >> u >> v;

		eg[u].push_back(v);

		eg[v].push_back(u);

		deg[u]++; deg[v]++;

	}

	sam.clear();

	for (int i = 1; i <= n; ++i)

		if (deg[i] == 1)

			dfs(i, 0, 0);

	sam.solve();

}

總結

以上是生活随笔為你收集整理的BZOJ3926 (后缀自动机)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。