后缀自动机1002 SPOJ NSUBSTR

最新推荐文章于 2019-10-04 03:05:32 发布

humeay

最新推荐文章于 2019-10-04 03:05:32 发布

阅读量352

点赞数

分类专栏：字符串

本文链接：https://blog.csdn.net/qq_27925701/article/details/52149232

版权

字符串专栏收录该内容

5 篇文章 0 订阅

订阅专栏

题意:
求长度为i的最多的出现次数
思路:
对于Sam上的每个节点保存匹配串的最长匹配长度
然后再传给父亲节点就行了

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<queue>
#include<stack>
#include<string>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define lowbit(x) (x&(-x))
typedef long long LL;
const int maxn = 250005;
const int inf=(1<<28)-1;
struct sam_node
{
    int fa,son[26];
    int len;
    void init(int _len)
    {
        len=_len;fa=-1;
        memset(son,-1,sizeof(son));
    }
};
class SAM
{
    public:
    int root,last,tots;
    sam_node sam[maxn*2];//length*2
    SAM()
    {
        init();
    }
    void init()
    {
        tots=0;
        root=last=0;
        sam[tots].init(0);
    }
    void extend(char ch)
    {
        int w=ch-'a';
        int p=last;
        int np=++tots;sam[tots].init(sam[p].len+1);
        int q,nq;while(p!=-1&&sam[p].son[w]==-1)
        sam[p].son[w]=np,p=sam[p].fa;
        if (p==-1) sam[np].fa=root;
        else
        {
            q=sam[p].son[w];
            if (sam[p].len+1==sam[q].len) sam[np].fa=q;
            else
            {
                nq=++tots;sam[nq].init(0);
                sam[nq]=sam[q];
                sam[nq].len=sam[p].len+1;
                sam[q].fa=nq;sam[np].fa=nq;
                while(p!=-1&&sam[p].son[w]==q)
                sam[p].son[w]=nq,p=sam[p].fa;
            }
        }
        last=np;
    }
    void build(char* str)
    {
        int len=strlen(str);
        for(int i=0;i<len;++i) extend(str[i]);
    }
    int r[maxn*2],w[maxn];
    //r[i]为topsort后 入度为0先入队 
    void topsort()
    {
        int i,len=tots;
        memset(w,0,sizeof(w));
        for(int i=1;i<=tots;++i) w[sam[i].len]++;
        for(int i=1;i<=len;++i) w[i]+=w[i-1];
        for(int i=tots;i>=1;--i) r[w[sam[i].len]--]=i;
        r[0]=0;
    }
    void output()
    {
        for(int i=0;i<=tots;++i)
        {
            printf("%d(fa=%d): ",i,sam[i].fa);
            for(int j=0;j<26;++j)
            if(sam[i].son[j]!=-1)
            printf("%c(%d) ",j+'a',sam[i].son[j]);
            printf("\n");
        }
    }
    int dp[maxn*2],Ans[maxn];
    void Solve(char* str)
    {
        topsort();
        //output();
        memset(dp,0,sizeof(dp));
        memset(Ans,0,sizeof(Ans));
        int len=strlen(str);
        int now=root;
        for(int i=0;i<len;++i)
        {
            now=sam[now].son[str[i]-'a'];
            dp[now]++;
        }
        for(int i=tots;i>=1;--i)
        {
            int p=r[i];
            dp[sam[p].fa]+=dp[p];
            Ans[sam[p].len]=max(Ans[sam[p].len],dp[p]);
        }
        for(int i=len-1;i>=1;--i)
            Ans[i]=max(Ans[i],Ans[i+1]);
        for(int i=1;i<=len;++i)
            printf("%d\n",Ans[i]);
    }
}Sam;
char str[maxn];
int main()
{
    scanf("%s",str);
    int len=strlen(str);
    for(int i=0;i<len;++i)
        Sam.extend(str[i]);
    Sam.Solve(str);
    return 0;
}