当前位置: 代码迷 >> 综合 >> 【后缀自动机】Hihocoder#1449 : 后缀自动机三·重复旋律6
  详细解决方案

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

热度:80   发布时间:2023-09-27 03:17:04.0

分析:

后缀自动机板子题。

维护endpos集合大小。

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

#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#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]);
}
  相关解决方案