首页 > 代码库 > SPOJ7258

SPOJ7258

传送门

这题可以参考平衡树求第k大的过程,需要预处理一下从当前节点往下走能走出多少个子串。

原本准备存个图用反向的topsort,发现极为麻烦,看了别人的代码后发现,他们按step大小用了基排,省了很多麻烦。

仔细看了一下构造过程,貌似确实有这样的性质,后创建且step大的节点在fail树上处于儿子的地位,在原图上处于被转移点的地位。

如果性质成立,那么这里也就说得通了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<set>
#include<map>
#include<queue>
using namespace std;
#define mem1(i,j) memset(i,j,sizeof(i))
#define mem2(i,j) memcpy(i,j,sizeof(i))
#define LL long long
#define up(i,j,n) for(int i=(j);i<=(n);i++)
#define FILE "dealing"
#define poi vec
#define eps 1e-10
#define db double 
const int maxn=500010,inf=1000000000,mod=1000000007;
int read(){
	int x=0,f=1,ch=getchar();
	while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
	while(ch<=‘9‘&&ch>=‘0‘){x=(x<<1)+(x<<3)+ch-‘0‘,ch=getchar();}
	return f*x;
}
bool cmax(int& a,int b){return a<b?a=b,true:false;}
bool cmin(int& a,int b){return a>b?a=b,true:false;}
char s[maxn];int f[maxn];
namespace SAM{
	int cnt=1,now=1;
	int pre[maxn],c[maxn][26],len[maxn],ct[maxn],sa[maxn];
	void extend(int x){
		int np,p,q,nq;
		p=now;np=++cnt;len[np]=len[p]+1;now=np;
		while(p&&!c[p][x])c[p][x]=np,p=pre[p];
		if(!p)pre[np]=1;
		else {
			q=c[p][x];
			if(len[q]==len[p]+1)pre[np]=q;
			else {
				len[nq=++cnt]=len[p]+1;
				mem2(c[nq],c[q]);
				pre[nq]=pre[q];
				pre[q]=pre[np]=nq;
				while(p&&c[p][x]==q)c[p][x]=nq,p=pre[p];
			}
		}
	}
	void getsort(){
		up(i,1,cnt)ct[len[i]]++;
		up(i,1,cnt)ct[i]+=ct[i-1];
		for(int i=cnt;i>=1;i--)sa[ct[len[i]]--]=i;
		for(int i=cnt;i>=1;i--){
			f[sa[i]]=1;
			up(j,0,25)f[sa[i]]+=f[c[sa[i]][j]];
		}
	}
	void build(char* s){
		cnt=1,now=1;int n=strlen(s+1);
		up(i,1,n)extend(s[i]-‘a‘);
	}
};
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	scanf("%s",s+1);
	SAM::build(s);
	SAM::getsort();
	int Q=read();
	while(Q--){
		int k=read();
		int now=1,len=0;
		while(k)up(i,0,25){
			if(SAM::c[now][i]){
				if(f[SAM::c[now][i]]>=k){
					putchar(‘a‘+i);
					now=SAM::c[now][i];
					k--;//不可省
					break;
				}
				else k-=f[SAM::c[now][i]];
			}
		}
		printf("\n");
	}
	return 0;
}

  

SPOJ7258