整數 字母 In 英文 out pan str spa test

BZOJ_3998_[TJOI2015]弦論_後綴自動機

Description

對於一個給定長度為N的字符串，求它的第K小子串是什麽。

Input

第一行是一個僅由小寫英文字母構成的字符串S

第二行為兩個整數T和K，T為0則表示不同位置的相同子串算作一個。T=1則表示不同位置的相同子串算作多個。K的意義如題所述。

Output

輸出僅一行，為一個數字串，為第K小的子串。如果子串數目不足K個，則輸出-1

Sample Input

aabc
0 3

Sample Output

aab

HINT

N<=5*10^5

T<2 K<=10^9

---

首先肯定是要一位一位確定字符，這樣我們就需要兩個標記。 建出後綴自動機。 設siz[i]表示i對應子串在串中出現了多少次，f[i]表示i對應子串有多少種轉移方法。 siz[i]比較好求，如果表示不同位置的相同子串算作一個所有siz都為1，否則siz可以通過後綴樹DP出來。 f[i]用拓撲序DP出來，初始值為siz。 然後按位確定每一位的字符即可。 代碼：

#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 1000050
int ch[N][26],fa[N],dep[N],siz[N],cnt=1,lst=1,f[N];
int ws[N],a[N];
char w[N];
void insert(int x) {
	int p=lst,np=++cnt,q,nq;
	lst=np; dep[np]=dep[p]+1;
	for(;p&&!ch[p][x];p=fa[p]) ch[p][x]=np;
	if(!p) fa[np]=1;
	else {
		q=ch[p][x];
		if(dep[q]==dep[p]+1) fa[np]=q;
		else {
			fa[nq=++cnt]=fa[q];
			dep[nq]=dep[p]+1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));
			fa[q]=fa[np]=nq;
			for(;p&&ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
		}
	}
	siz[np]=1;
}
void print() {
	int i,j;
	printf("test-------------------------------------------\n");
	for(i=1;i<=cnt;i++) {
		printf("p=%d,siz=%d,dep=%d,fa=%d\n",i,siz[i],dep[i],fa[i]);
		for(j=0;j<26;j++) {
			if(ch[i][j]) {
				printf("ch(%d)(%c)=%d\n",i,j+‘a‘,ch[i][j]);
			}
		}
	}
	printf("lst=%d\n",lst);
}
int main() {
	int T,K;
	scanf("%s%d%d",w+1,&T,&K);
	int n=strlen(w+1);
	int i,j,p;
	for(i=1;i<=n;i++) insert(w[i]-‘a‘);
	for(i=1;i<=cnt;i++) ws[dep[i]]++;
	for(i=1;i<=n;i++) ws[i]+=ws[i-1];
	for(i=cnt;i;i--) a[ws[dep[i]]--]=i;
	if(T) {
		for(i=cnt;i;i--) {
			p=a[i]; siz[fa[p]]+=siz[p];
		}
	}else for(i=cnt;i;i--) siz[a[i]]=1;
	for(siz[1]=0,i=cnt;i;i--) {
		p=a[i];
		for(f[p]=siz[p],j=0;j<26;j++) f[p]+=f[ch[p][j]];
	}
	if(f[1]<K) puts("-1");
	else {
		p=1;
		while(K>siz[p]) {
			K-=siz[p];
			for(i=0;i<26;i++) {
				if(K<=f[ch[p][i]]) {printf("%c",i+‘a‘),p=ch[p][i]; break;}
				else K-=f[ch[p][i]];
			}
		}
		puts("");
	}
}

BZOJ_3998_[TJOI2015]弦論_後綴自動機