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[N][26], fa[N], ma[N], v[N], s[N], b[N], 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