SPOJ 7258 (后缀自动机)

转载:http://hzwer.com/4492.html

给一个长度不超过90000的串S，每次询问它的所有不同子串中，字典序第K小的，询问不超过500个。

搞出后缀自动机

dp处理出每个点往下走能够走出多少个串。
f[i]=sigma(f[ch[i][c])+1
这个可以按Max排序之后倒着推就好了。
询问的时候看一下走下去个数是否<=k，是的话就走下去，然后–k；否则就找下一条边

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#define N 180005
#define inf 1000000000
using namespace std;
char ch[N];
int Q,p,q,np,nq;
int cnt=1,last=1,n;
int fa[N],to[N][26],l[N],s[N];
int f[N],v[N];
void extend(int c)
{
    p=last;np=last=++cnt;l[np]=++n;
    for(;p&&!to[p][c];p=fa[p])to[p][c]=np;
    if(!p)fa[np]=1;
    else 
    {
        int q=to[p][c];
        if(l[p]+1==l[q])fa[np]=q;
        else 
        {
            nq=++cnt;l[nq]=l[p]+1;
            memcpy(to[nq],to[q],sizeof(to[q]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            for(;to[p][c]==q;p=fa[p])to[p][c]=nq;
        }
    }
}
void build()
{
    scanf("%s",ch);
    for(char *c=ch;*c;c++)extend(*c-'a');    
}
void pre()
{
    for(int i=1;i<=cnt;i++)v[l[i]]++;
    for(int i=1;i<=n;i++)v[i]+=v[i-1];
    for(int i=1;i<=cnt;i++)s[v[l[i]]--]=i;
    for(int i=cnt;i;i--)
    {
        f[s[i]]=1;
        for(int j=0;j<26;j++)
            f[s[i]]+=f[to[s[i]][j]];
    }
}
void query()
{
    p=1;int x;
    scanf("%d",&x);
    while(x)
    {
        for(int i=0;i<26;i++)
            if(to[p][i])
            {
                if(f[to[p][i]]>=x)
                {
                    putchar('a'+i);
                    p=to[p][i];
                    --x;break;
                }
                else x-=f[to[p][i]];
            }
    }
    printf("
");
}
int main()
{
    build();
    pre();
    scanf("%d",&Q);
    while(Q--)
        query();
    return 0;
}