【TJOI2015】【BZOJ3998】弦论

来源：互联网发布：linux date修改时区编辑：程序博客网时间：2024/06/08 07:52

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;                }            }        }    }}

1 0