【TJOI2015】【BZOJ3998】弦论

Description

Solution

后缀三姐妹都可以做这题。
我选择最短的。

后缀自动机

不会的参照后缀自动机学习小记
建出后缀自动机。
先对后缀自动机拓扑一下（其实是为了方便求构出的字符串的个数和right集合的个数），其实不用真的求拓扑序，把len排个序就好了，很显然len的顺序就是拓扑序。
然后求出当前这个状态right的集合的大小（表示这个状态的字符串出现了多少次）。
那么求答案一个dfs就可以了，但是可能会爆栈，所以打成非递归版的就可以了。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int maxn=500007;
int i,j,l,tt,n,m,T,ans1,ans2,k;
int last,num;
struct node{
    int fa,len,sum;
    int son[26];
}t[maxn*2];
char s[maxn*2];
int a[maxn*2],b[maxn*2],cc[maxn*2],p,np,q,nq;
void extend(int c){
    p=last,np=++num;
    t[np].len=t[p].len+1;np=num;cc[np]=1;
    while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].fa;
    if(!p)t[np].fa=1;
    else{
        q=t[p].son[c];
        if(t[p].len+1==t[q].len)t[np].fa=q;
        else{
            nq=++num;
            t[nq]=t[q];
            t[num].len=t[p].len+1;
            t[q].fa=t[np].fa=nq;
            while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].fa;
        }
    }
    last=np;
}
int main(){
    freopen("string.in","r",stdin);
    freopen("string.out","w",stdout);
//  freopen("fan.in","r",stdin);
//  freopen("fan.out","w",stdout);
    scanf("%s",s+1);
    scanf("%d%d",&tt,&k);
    n=strlen(s+1);num=last=1;
    fo(i,1,n)extend(s[i]-'a');
    fo(i,1,num)a[t[i].len]++;
    fo(i,1,n)a[i]+=a[i-1];
    fod(i,num,1)b[a[t[i].len]--]=i;
    fod(i,num,1){
        int u=b[i];
        if(tt)cc[t[u].fa]+=cc[u];
        else cc[u]=1;
    }
    cc[1]=0;
    fod(i,num,1){
        int u=b[i];t[u].sum=cc[u];
        fo(j,0,25){
            t[u].sum+=t[t[u].son[j]].sum;
        }
    }
    if(t[1].sum<k)printf("-1\n");
    else{
        int x=1;
        while(1){
            if(k<=cc[x])return 0;
            k-=cc[x];
            fo(i,0,25){
                if(t[x].son[i]){
                    if(k<=t[t[x].son[i]].sum){
                        putchar('a'+i);
                        x=t[x].son[i];
                        break;
                    }
                    k-=t[t[x].son[i]].sum;
                }
            }
        }
    }
}