bzoj3998 [TJOI2015]弦论（SA/SAM）

求第k小子串。如果t=0，直接拿sa线性搞即可。如果t=1，我复杂度就炸了。。。然而本机和luogu都跑了过去（可见数据之水，逃）。正解大概是SAM，待更新。

3.24upd:蒟蒻我终于看懂了SAM的板子orz
顾名思义，SAM就是能识别所有后缀的自动机。我们有一个朴素的构建SAM的想法：把每一个后缀都插到Trie里去！但是这样是O(n^2)的，无论时间还是空间都会炸。

怎么办呢？我们要压缩状态，一些子串是可以压在一起的，具体怎么搞可以见cljWC2012讲稿：portal。Candy？的总结也蛮不错的：portal

我们不仅要搞出SAM，还要搞出Parent Tree，其实很多时候我们都是在利用Parent Tree的性质。注意节点是O(2n)的。

把所有节点按mx排序（利用基数排序可以做到O(n)），可以得到Parent Tree及SAM的拓扑序，我们就可以倒着计算出每个节点的|Right|了，或者dp搞些什么事情。总的复杂度都是O(n)的。

对于此题，我们还要处理出sum[p]数组来方便的查询第K小。

SA

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 500010
#define ll long long
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int type,n,K,sa[N],rk[N<<1],rk1[N],tmp[N],cnt[N],h[N],m='z'+1;
char s[N];
int main(){
//  freopen("string.in","r",stdin);
//  freopen("string.out","w",stdout);
    scanf("%s",s+1);n=strlen(s+1);type=read();K=read();
    for(int i=1;i<=n;++i) cnt[s[i]]=1;
    for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
    for(int i=1;i<=n;++i) rk[i]=cnt[s[i]];
    int k=0;
    for(int p=1;k!=n;p+=p,m=k){
        for(int i=1;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[i+p]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) tmp[cnt[rk[i+p]]--]=i;
        for(int i=1;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[tmp[i]]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) sa[cnt[rk[tmp[i]]]--]=tmp[i];
        memcpy(rk1,rk,sizeof(rk1));rk[sa[1]]=k=1;
        for(int i=2;i<=n;++i){
            if(rk1[sa[i]]!=rk1[sa[i-1]]||rk1[sa[i]+p]!=rk1[sa[i-1]+p]) ++k;
            rk[sa[i]]=k;
        }
    }k=0;
    for(int i=1;i<=n;++i){
        if(rk[i]==1){h[1]=0;continue;}
        if(i==1||h[rk[i-1]]<=1) k=0;
        if(k) --k;
        while(s[i+k]==s[sa[rk[i]-1]+k]) ++k;
        h[rk[i]]=k;
    }for(int i=1;i<=n;++i){
        if(type){
            for(int j=h[i]+1;sa[i]+j-1<=n;++j){
                int r=i+1;
                while(h[r]>=j) ++r;--r;
                if(K<=r-i+1){
                    for(int k=1;k<=j;++k) putchar(s[sa[i]+k-1]);return 0;
                }K-=r-i+1;  
            }continue;
        }int cnt=n-(sa[i]+h[i])+1;if(K>cnt){K-=cnt;continue;}
        for(int j=1;j<=K+h[i];++j) putchar(s[sa[i]+j-1]);return 0;
    }puts("-1");return 0;
}

SAM

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 1000010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
char s[N];
int n=0,m,son[N][26],par[N],mx[N],val[N],sum[N],last,rt,type,K;
int cnt[N],a[N];//sum[p] 就是到达p节点以后能识别的串的个数  val[p]就是p点代表状态的出现次数 
inline void ins(int ch){
    int p=last,np=++n;last=np;mx[np]=mx[p]+1;val[np]=1;
    for(;p&&!son[p][ch];p=par[p]) son[p][ch]=np;
    if(!p){par[np]=rt;return;}
    int q=son[p][ch];
    if(mx[q]==mx[p]+1){par[np]=q;return;}
    int nq=++n;mx[nq]=mx[p]+1;memcpy(son[nq],son[q],sizeof(son[q]));
    par[nq]=par[q];par[q]=par[np]=nq;
    for(;p&&son[p][ch]==q;p=par[p]) son[p][ch]=nq;
}
void dfs(int x){
    if(K<=val[x]) return;
    K-=val[x];
    for(int j=0;j<26;++j){
        int y=son[x][j];
        if(K<=sum[y]){putchar(j+'a');dfs(y);return;}
        K-=sum[y];
    }
}
int main(){
//  freopen("a.in","r",stdin);
    scanf("%s",s+1);m=strlen(s+1);last=rt=++n;
    for(int i=1;i<=m;++i) ins(s[i]-'a');
    type=read();K=read();
    for(int i=1;i<=n;++i) cnt[mx[i]]++;
    for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
    for(int i=n;i>=1;--i) a[cnt[mx[i]]--]=i;
    for(int i=n;i>=1;--i){
        int p=a[i];type?val[par[p]]+=val[p]:val[p]=1;
    }val[1]=0;
    for(int i=n;i>=1;--i){
        int p=a[i];sum[p]=val[p];
        for(int j=0;j<26;++j) sum[p]+=sum[son[p][j]];
    }if(K>sum[rt]) puts("-1");
    else dfs(rt);
    return 0;
}