BZOJ 3998 TJOI 2015 弦论 后缀自动机

求第k大子串

http://blog.csdn.net/huanghongxun/article/details/51164051

#include <cstring>
#include <cstdio>
#define FOR(i,j,k) for(i=j;i<=k;++i)
const int rt = 1, N = 1000005;
int last = 1, cnt = 1, len = 0;
int ch
[26], fa
, ma
, v
, s
, b
, c[N / 2];
char str[N / 2];
void add(char c) {
int np = ++cnt, p = last, q, nq; last = np; v[np] = 1; ma[np] = ma[p] + 1;
while (p && !ch[p][c]) ch[p][c] = np, p = fa[p];
if (!p) fa[np] = rt;
else {
q = ch[p][c];
if (ma[q] == ma[p] + 1) fa[np] = q;
else {
nq = ++cnt; memcpy(ch[nq], ch[q], sizeof ch[q]);
ma[nq] = ma[p] + 1; fa[nq] = fa[q]; fa[q] = fa[np] = nq;
while (p && ch[p][c] == q) ch[p][c] = nq, p = fa[p];
}
}
}
int main() {
int i, j, n, q, t, k, p = rt;
scanf("%s%d%d", str, &t, &k);
n = strlen(str);
for (i = 0; str[i]; ++i) add(str[i] - 'a');
FOR(i,1,cnt) ++c[ma[i]];
FOR(i,1,n) c[i] += c[i - 1];
for (i = cnt; i; --i) b[c[ma[i]]--] = i;
for (i = cnt; i; --i)
if (t == 0) v[b[i]] = 1;
else v[fa[b[i]]] += v[b[i]];
v[rt] = 0;
for (i = cnt; i; --i) {
s[b[i]] = v[b[i]];
FOR(j,0,25) s[b[i]] += s[ch[b[i]][j]];
}
if (k > s[rt]) return puts("-1"), 0;
while (k) {
if (k <= v[p]) break;
k -= v[p];
FOR(i,0,25) if (q = ch[p][i])
if (k > s[q]) k -= s[q];
else {
putchar(i + 'a');
p = ch[p][i];
break;
}
}
return 0;
}

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