#include <stdio.h>
int main()
{
puts("转载请注明出处[vmurder]谢谢");
puts("网址:blog.csdn.net/vmurder/article/details/45369569");
}
首先我们可以建一个后缀自动机。
然后每条路径走到每个点都是一个串,它们是有字典序的。
我们只需要统计出往每个点走之后都有多少串就好了。
对于不计重复的情况下,
对于计算重复的情况下,每个节点都有多种走到最后的方式,
比如
初值是所有
每次往下走就行了,类似26分? (雾
不妨看我的
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 1001000
#define T 26
#define inf 0x3f3f3f3f
using namespace std;
struct SAM
{
int son[N][T],pa[N],dep[N],last,cnt;
int val[N],f[N];
void init(){last=cnt=1;}
int newnode(int p){dep[++cnt]=dep[p]+1;return cnt;}
void add(int x)
{
int p=last;
int np=newnode(p);
while(p&&!son[p][x])son[p][x]=np,p=pa[p];
if(!p)pa[np]=1;
else {
int q=son[p][x];
if(dep[q]==dep[p]+1)pa[np]=q;
else {
int nq=newnode(p);
pa[nq]=pa[q],pa[q]=pa[np]=nq;
memcpy(son[nq],son[q],sizeof son[q]);
while(p&&son[p][x]==q)son[p][x]=nq,p=pa[p];
}
}
val[last=np]=1;
}
int d[N],stk[N],top;
struct Eli
{
int v,next;
}e[N*T/5];
int head[N],ct;
void add(int u,int v)
{
d[v]++;
e[++ct].v=v;
e[ct].next=head[u];
head[u]=ct;
}
void bfsf()
{
int i,j,u,v;
for(i=1;i<=cnt;i++)for(j=0;j<T;j++)if(son[i][j])
add(son[i][j],i);
for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
while(top)
{
u=stk[top--],f[u]++,val[u]=1;
for(i=head[u];i;i=e[i].next)
{
v=e[i].v,f[v]+=f[u];
if(!--d[v])stk[++top]=v;
}
}
f[1]--;
}
void bfsg()
{
int i,j,u,v;
for(i=2;i<=cnt;i++)d[pa[i]]++;
for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
while(top)
{
u=stk[top--];
val[pa[u]]+=val[u];
if(!--d[pa[u]])stk[++top]=pa[u];
}
for(i=1;i<=cnt;i++)for(j=0;j<T;j++)
if(son[i][j])add(son[i][j],i);
for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
while(top)
{
u=stk[top--],f[u]+=val[u];
for(i=head[u];i;i=e[i].next)
{
v=e[i].v,f[v]+=f[u];
if(!--d[v])stk[++top]=v;
}
}
f[1]-=val[1];
}
void dfs(int x,int k)
{
int i,v;
for(i=0;i<T;i++)if(v=son[x][i])
{
if(k<=f[v])
{
printf("%c",‘a‘+i);
k-=val[v];
if(k>0)dfs(v,k);
return ;
}
else k-=f[v];
}
}
}sam;
char s[N/2];
int main()
{
freopen("test.in","r",stdin);
int i,j,k;
scanf("%s",s+1);
sam.init();
for(i=1;s[i];i++)sam.add(s[i]-‘a‘);
scanf("%d%d",&j,&k);
if(j==0)sam.bfsf();
else sam.bfsg();
sam.dfs(1,k);
return 0;
}
原文地址:http://blog.csdn.net/vmurder/article/details/45369569