【BZOJ3998】【TJOI2015】弦论 后缀自动机

链接：

#include <stdio.h>
int main()
{
    puts("转载请注明出处[vmurder]谢谢");
    puts("网址：blog.csdn.net/vmurder/article/details/45369569");
}

题解：

首先我们可以建一个后缀自动机。
然后每条路径走到每个点都是一个串，它们是有字典序的。
我们只需要统计出往每个点走之后都有多少串就好了。
$f_i = (\sum {f_{son}})+num_i$
对于不计重复的情况下，$num_i=1$
对于计算重复的情况下，每个节点都有多种走到最后的方式，$num_i$ 就是看有这个种数。
比如 $ababab$，$aba$ 这个串最后可以走成 $abab$、$ababab$ 。于是 $num$ 是2。
初值是所有 $np$ 的 $num=1$ ，然后 $num_i = \sum {num_{son}}$ 。

$f函数$ 处理出来以后，就随便搞了。
每次往下走就行了，类似26分?　(雾
不妨看我的 $dfs$ 函数。

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 1001000
#define T 26
#define inf 0x3f3f3f3f
using namespace std;
struct SAM
{
    int son[N][T],pa[N],dep[N],last,cnt;
    int val[N],f[N];
    void init(){last=cnt=1;}
    int newnode(int p){dep[++cnt]=dep[p]+1;return cnt;}
    void add(int x)
    {
        int p=last;
        int np=newnode(p);
        while(p&&!son[p][x])son[p][x]=np,p=pa[p];
        if(!p)pa[np]=1;
        else {
            int q=son[p][x];
            if(dep[q]==dep[p]+1)pa[np]=q;
            else {
                int nq=newnode(p);
                pa[nq]=pa[q],pa[q]=pa[np]=nq;
                memcpy(son[nq],son[q],sizeof son[q]);
                while(p&&son[p][x]==q)son[p][x]=nq,p=pa[p];
            }
        }
        val[last=np]=1;
    }
    int d[N],stk[N],top;
    struct Eli
    {
        int v,next;
    }e[N*T/5];
    int head[N],ct;
    void add(int u,int v)
    {
        d[v]++;
        e[++ct].v=v;
        e[ct].next=head[u];
        head[u]=ct;
    }
    void bfsf()
    {
        int i,j,u,v;
        for(i=1;i<=cnt;i++)for(j=0;j<T;j++)if(son[i][j])
                add(son[i][j],i);
        for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
        while(top)
        {
            u=stk[top--],f[u]++,val[u]=1;
            for(i=head[u];i;i=e[i].next)
            {
                v=e[i].v,f[v]+=f[u];
                if(!--d[v])stk[++top]=v;
            }
        }
        f[1]--;
    }
    void bfsg()
    {
        int i,j,u,v;
        for(i=2;i<=cnt;i++)d[pa[i]]++;
        for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
        while(top)
        {
            u=stk[top--];
            val[pa[u]]+=val[u];
            if(!--d[pa[u]])stk[++top]=pa[u];
        }
        for(i=1;i<=cnt;i++)for(j=0;j<T;j++)
            if(son[i][j])add(son[i][j],i);
        for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
        while(top)
        {
            u=stk[top--],f[u]+=val[u];
            for(i=head[u];i;i=e[i].next)
            {
                v=e[i].v,f[v]+=f[u];
                if(!--d[v])stk[++top]=v;
            }
        }
        f[1]-=val[1];
    }
    void dfs(int x,int k)
    {
        int i,v;
        for(i=0;i<T;i++)if(v=son[x][i])
        {
            if(k<=f[v])
            {
                printf("%c",'a'+i);
                k-=val[v];
                if(k>0)dfs(v,k);
                return ;
            }
            else k-=f[v];
        }
    }
}sam;
char s[N/2];
int main()
{
    freopen("test.in","r",stdin);
    int i,j,k;

    scanf("%s",s+1);
    sam.init();
    for(i=1;s[i];i++)sam.add(s[i]-'a');
    scanf("%d%d",&j,&k);
    if(j==0)sam.bfsf();
    else sam.bfsg();
    sam.dfs(1,k);
    return 0;
}