【hihoCoder 1449】後綴自動機 SAM

題目很好懂

分析:
類似於洛谷P3804,這個題目只需要用一個map在dfs parent tree的時候記錄每個長度的siz的最大值,然後直接輸出就做完了。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e6+7;
struct node
{
	int ch[26];
	int len,fa;
}point[maxn<<1];
int las = 1,tot = 1;
int siz[maxn<<1];
vector<int> G[maxn<<1];
unordered_map<int,long long> mp;
void add(char c)
{
	int p = las;
	int np = las = ++tot;
	siz[tot] = 1;
	point[np].len = point[p].len+1;
	for(;p && !point[p].ch[c];p=point[p].fa) point[p].ch[c] = np;
	if(!p) point[np].fa = 1;
	else
	{
		int q = point[p].ch[c];
		if(point[q].len == point[p].len+1) point[np].fa = q;
		else
		{
			int nq = ++tot;
			point[nq] = point[q];
			point[nq].len = point[p].len+1;
			point[q].fa = point[np].fa = nq;
			for(;p && point[p].ch[c]==q;p=point[p].fa) point[p].ch[c] = nq;
		}
	}
}
void dfs(int s)
{
	for(int i=0;i<G[s].size();i++)
	{
		int v = G[s][i];
		dfs(v);
		siz[s] += siz[v];
	}
	mp[point[s].len] = max(mp[point[s].len],1LL*siz[s]);
}
int main()
{
	string s;
	cin>>s; 
	for(int i=0;i<s.size();i++) add(s[i]-'a');
	for(int i=2;i<=tot;i++) G[point[i].fa].push_back(i);
	dfs(1);
	for(int i=1;i<=s.size();i++) printf("%lld\n",mp[i]);
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章