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;
}