SPOJ

感觉后缀自动机好难啊，比后缀数组难多了...看了好几天勉强懂了一丢丢

学习博客：

洛谷某大佬

HihoCoder

题目传送门：https://www.spoj.com/problems/SUBLEX/en/

题意：给你一个字符串，然后是q次查询，每次输出第i小的字符串。

题解：先构造出后缀自动机，然后对每个节点预处理出从这个节点可以到达多少个不同的子串，然后沿着SAM一路查找下去并更新。

#include<bits/stdc++.h>
using namespace std;

const int maxn=2e6+10;

struct state
{
    int mxlen,link;
    int cnt;
    int nt[26];
}st[maxn<<1];

int sz,last;
char s[maxn];

inline void sam_init()
{
   st[0].mxlen=0;
   st[0].link=-1;
   sz=1;
   last=0;
}

inline void sam_extend(int c)
{
    c-='a';
    int cur=sz++;
    st[cur].mxlen=st[last].mxlen+1;
    st[cur].cnt=1;
    int p=last;
    while(p!=-1&&!st[p].nt[c]){
        st[p].nt[c]=cur;
        p=st[p].link;
    }
    if(p==-1) st[cur].link=0;
    else{
        int q=st[p].nt[c];
        if(st[p].mxlen+1==st[q].mxlen) st[cur].link=q;
        else{
            int clone=sz++;
            st[clone].mxlen=st[p].mxlen+1;
            memcpy(st[clone].nt,st[q].nt,sizeof(st[q].nt));
            st[clone].link=st[q].link;
            while(p!=-1&&st[p].nt[c]==q){
                st[p].nt[c]=clone;
                p=st[p].link;
            }
            st[q].link=st[cur].link=clone;
        }
    }
    last=cur;
}

int b[maxn],c[maxn];

void build()        //这个难道不是基排吗，为啥看好多博客写拓扑排序,,,
{
    int len=strlen(s+1);
    for(int i=1;i<=len;i++) sam_extend(s[i]);
    for(int i=1;i<=sz;i++) c[st[i].mxlen]++;
    for(int i=1;i<=sz;i++) c[i]+=c[i-1];
    for(int i=1;i<=sz;i++) b[c[st[i].mxlen]--]=i;
    for(int i=sz;i>=1;i--){
        st[b[i]].cnt=1;
        for(int j=0;j<26;j++)
            st[b[i]].cnt+=st[st[b[i]].nt[j]].cnt;     //st[].cnt记录当前节点不同子串个数
    }
}

void query(int k)
{
    int p=0;
    while(k){
        for(int i=0;i<26;i++){
            if(st[p].nt[i]){
                if(k>st[st[p].nt[i]].cnt) k-=st[st[p].nt[i]].cnt;
                else {
                    char ch=i+'a';
                    cout<<ch;
                    p=st[p].nt[i];
                    k--;
                    break;
                }
            }
        }
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin>>s+1;
    sam_init();
    build();
    int m;
    cin>>m;
    while(m--){
        int n;
        cin>>n;
        query(n);
        cout<<endl;
    }
    return 0;
}

有一个地方我一直没搞清楚，这两种写法不一样吗？一种是父亲加上他，一种是它加儿子，最后结果应该是一样的吧...但是第一种写法是错的。

    for(int i=sz;i>=1;i--) st[i].cnt=1;
    for(int i=sz;i>=1;i--) st[st[b[i]].link].cnt+=st[b[i]].cnt;

1　　 for(int i=sz;i>=1;i--){
        st[b[i]].cnt=1;
        for(int j=0;j<26;j++)
            st[b[i]].cnt+=st[st[b[i]].nt[j]].cnt;
    }