【spoj8222】Substrings 后缀自动机

AC通道：http://www.spoj.com/problems/NSUBSTR/

【题解】

我们知道，在SAM上对于一个结点s,Max(s)的出现次数就是right(s)

那么问题就是我们如何求出每个点的right集的大小。

我们知道，一个点的right集为它的儿子的right集的并集，所以我们按照拓扑序遍历每个点，同时更新它父亲的答案。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
#define FILE "read"
#define MAXN 500010
#define up(i,j,n) for(int i=j;i<=n;++i)
#define dn(i,j,n) for(int i=j;i>=n;--i)
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
int n,cnt(1),now(1),f[MAXN],mx[MAXN],c[MAXN],id[MAXN],g[MAXN],ans[MAXN],son[MAXN][27];
char ch[MAXN];
void insert(int x){
	int p=now,np=++cnt;
	mx[np]=mx[now]+1;now=np;g[np]=1;
	while(!son[p][x]&&p) son[p][x]=np,p=f[p];
	if(!p) f[np]=1;
	else{
		int q=son[p][x];
		if(mx[q]==mx[p]+1) f[np]=q;
		else{
			int nq=++cnt;
			mx[nq]=mx[p]+1;
			memcpy(son[nq],son[q],sizeof(son[q]));
			f[nq]=f[q];  f[q]=f[np]=nq;
			while(son[p][x]==q&&p) son[p][x]=nq,p=f[p];
		}
	}
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	scanf("%s",ch+1); n=strlen(ch+1);
	up(i,1,n)    insert(ch[i]-'a');
	up(i,2,cnt)  c[mx[i]]++;
	up(i,1,n)    c[i]+=c[i-1];
	up(i,2,cnt)  id[c[mx[i]]--]=i;
	dn(i,cnt,2)  g[f[id[i]]]+=g[id[i]];
	up(i,2,cnt)  cmax(ans[mx[i]],g[i]);
	up(i,1,n)  printf("%d\n",ans[i]);
	return 0;
}