【后缀自动机】Hihocoder#1449 : 后缀自动机三·重复旋律6

分析:

后缀自动机板子题。

维护endpos集合大小。

复制出来的点不代表任何一个前缀,所以其endpos集合大小为所有fail指向它的endpos之和,否则其endpos集合大小为所有fail指向它的endpos之和+1

#include
#include
#include
#include
#define SF scanf
#define PF printf
#define MAXN 1000010
#define MAXM 1500010
using namespace std;
char s[MAXN];
struct node{
	node *ch[26];
	node *fail;
	vector<node *> son;
	int maxlen,siz;	
	bool flag;
}SuA[MAXM];
node *las=SuA,*ncnt=SuA,*rt=SuA;
void Ins(int c){
	node *p=las;
	node *np=++ncnt;
	np->flag=1;
	np->maxlen=p->maxlen+1;
	las=np;
	while(p&&p->ch[c]==NULL)
		p->ch[c]=np,p=p->fail;	
	if(!p)
		np->fail=rt;
	else{
		node *q=p->ch[c];
		if(q->maxlen==p->maxlen+1)
			np->fail=q;
		else{
			node *nq=++ncnt;
			*nq=*q;
			nq->maxlen=p->maxlen+1;
			q->fail=np->fail=nq;
			while(p&&p->ch[c]==q)
				p->ch[c]=nq,p=p->fail;
		}
	}
}
int ans[MAXN];
void dfs(node *x){
	x->siz=x->flag;
	for(int i=0;i<int(x->son.size());i++){
		node *u=x->son[i];
		dfs(u);
		x->siz+=u->siz;
	}
	ans[x->maxlen]=max(ans[x->maxlen],x->siz);
}
int main(){
	SF("%s",s);	
	int n=strlen(s);
	for(int i=0;i<n;i++)
		Ins(s[i]-'a');
	for(node *i=SuA+1;i<=ncnt;i++){
//		PF("%d %d %d\n",i-SuA,i->fail-SuA,i->flag);
		i->fail->son.push_back(i);
	}
	dfs(rt);
	for(int i=n-1;i>=1;i--)
		ans[i]=max(ans[i],ans[i+1]);
	for(int i=1;i<=n;i++)
		PF("%d\n",ans[i]);
}

你可能感兴趣的:(后缀自动机,字符串)