BZOJ 3998: [TJOI2015]弦论 后缀自动机

3998: [TJOI2015]弦论

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 4251  Solved: 1562
[Submit][Status][Discuss]

Description

对于一个给定长度为N的字符串，求它的第K小子串是什么。

Input

 第一行是一个仅由小写英文字母构成的字符串S

第二行为两个整数T和K，T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。

Output

输出仅一行，为一个数字串，为第K小的子串。如果子串数目不足K个，则输出-1

Sample Input

aabc
0 3

Sample Output

aab

给一个串，求第k小的字串，后两种情况，分别是相同字串不同位置不同和算相同

 

对于后缀自动机的每个节点求包括他本身和儿子的节点数。

如果相同字串算不同，那么对于每个节点，其祖先的个数需要加一。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxm = 1000005;
struct node
{
	int link, len, val, Next[26];
}st[maxm];
int son[maxm], vis[maxm], last, cnt, sum;
char str[maxm];
void init()
{
	memset(vis, 0, sizeof(vis));
	memset(st, 0, sizeof(st));
	st[0].link = -1, last = 0;
	cnt = 1, sum = 0;
}
void insert(int c,int flag)
{
	int p = last, now = cnt++, rev;
	st[now].len = st[p].len + 1;
	st[now].val = 1;
	while (p != -1 && !st[p].Next[c])
		st[p].Next[c] = now, p = st[p].link;
	if (p == -1) st[now].link = 0;
	else
	{
		int q = st[p].Next[c];
		if (st[q].len == st[p].len + 1)
			st[now].link = q;
		else
		{
			rev = cnt++;
			st[rev] = st[q];
			st[rev].len = st[p].len + 1;
			st[rev].val = st[q].val;
			while (p != -1 && st[p].Next[c] == q)
				st[p].Next[c] = rev, p = st[p].link;
			st[q].link = st[now].link = rev;
		}
	}
	if (flag) for (p = st[now].link;p != -1;p = st[p].link) st[p].val++;
	last = now;
}
void dfs(int k)
{
	int i, v;
	vis[k] = 1;
	son[k] = st[k].val;
	for (int i = 0;i < 26;i++)
	{
		v = st[k].Next[i];
		if (!v) continue;
		if (!vis[v])
			dfs(v);
		son[k] += son[v];
	}
}
void print(int x,int k)
{
	int i, v;
	if (!k) return;
	for (i = 0;i < 26;i++)
	{
		v = st[x].Next[i];
		if (!v) continue;
		if (k > son[v]) k -= son[v];
		else
		{
			print(v, k - st[v].val);
			return;
		}
	}
}
int main()
{
	int n, i, j, k;
	init();
	scanf("%s", str);
	scanf("%d%d", &n, &k);
	for (i = 0;str[i] != '\0';i++)
		insert(str[i] - 'a', n);
	dfs(0);
	if (son[0] < k) printf("-1");
	else print(0, k);
	putchar('\n');
	return 0;
}