SPOJ 7258 (后缀自动机)

时间:2023-03-10 05:27:52
SPOJ 7258 (后缀自动机)

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

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

搞出后缀自动机

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


 #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=,last=,n;
int fa[N],to[N][],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]=;
else
{
int q=to[p][c];
if(l[p]+==l[q])fa[np]=q;
else
{
nq=++cnt;l[nq]=l[p]+;
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=;i<=cnt;i++)v[l[i]]++;
for(int i=;i<=n;i++)v[i]+=v[i-];
for(int i=;i<=cnt;i++)s[v[l[i]]--]=i;
for(int i=cnt;i;i--)
{
f[s[i]]=;
for(int j=;j<;j++)
f[s[i]]+=f[to[s[i]][j]];
}
}
void query()
{
p=;int x;
scanf("%d",&x);
while(x)
{
for(int i=;i<;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("\n");
}
int main()
{
build();
pre();
scanf("%d",&Q);
while(Q--)
query();
return ;
}