先构建出原串的后缀自动机。
然后在后缀自动机上通过拓扑排序,预处理出每一条边能到达的字符串个数(如果是位置不同则初始贡献为
Right
R
i
g
h
t
集大小,本质不同则为
1
1
)。
从初始状态开始,枚举转移从到
z
z
<script type="math/tex" id="MathJax-Element-343">z</script>,利用贪心的思想尝试往下转移。
代码:
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 1e6 + 5;
struct cyx {
int fa, maxl, ri, go[26]; ll cnt;
void init() {
fa = maxl = ri = cnt = 0;
memset(go, 0, sizeof(go));
}
} T[N];
int n, QAQ, lst, op, K, que[N], w[N], tot;
char s[N], ans[N];
void ins(char ch) {
int c = ch - 'a', i = lst; T[lst = ++QAQ].init(); T[lst].ri = 1;
T[lst].maxl = T[i].maxl + 1; for (; i && !T[i].go[c]; i = T[i].fa)
T[i].go[c] = lst;
if (!i) T[lst].fa = 1; else {
int j = T[i].go[c];
if (T[i].maxl + 1 == T[j].maxl) T[lst].fa = j;
else {
int p; T[p = ++QAQ] = T[j]; T[p].ri = 0;
T[lst].fa = T[j].fa = p; T[p].maxl = T[i].maxl + 1;
for (; i && T[i].go[c] == j; i = T[i].fa)
T[i].go[c] = p;
}
}
}
void orzdalao() {
int i, c; for (i = 1; i <= QAQ; i++) w[T[i].maxl]++;
for (i = 1; i <= n; i++) w[i] += w[i - 1];
for (i = 1; i <= QAQ; i++) que[w[T[i].maxl]--] = i;
for (i = QAQ; i; i--) T[T[que[i]].fa].ri += T[que[i]].ri;
for (i = QAQ; i; i--) {
int u = que[i]; T[u].cnt = op ? T[u].ri : 1;
for (c = 0; c < 26; c++) T[u].cnt += T[T[u].go[c]].cnt;
}
}
void Run() {
int i; K += op ? T[1].ri : 1; ll orz = 0; for (i = 2; i <= QAQ; i++)
orz += op ? 1ll * T[i].ri * (T[i].maxl - T[T[i].fa].maxl) :
T[i].maxl - T[T[i].fa].maxl;
if (orz < K) return (void) puts("-1"); int c, u = 1; while (1) {
bool dl = 0; if (K == 1) break; K -= op ? T[u].ri : 1;
for (c = 0; c < 26; c++) {
int v = T[u].go[c]; if (!v) continue;
if (K <= T[v].cnt) {dl = 1; u = v; ans[++tot] = c + 'a'; break;}
else K -= T[v].cnt;
}
if (!dl) break;
}
for (i = 1; i <= tot; i++) printf("%c", ans[i]); cout << endl;
}
int main() {
int i; scanf("%s", s + 1); n = strlen(s + 1);
cin >> op >> K; T[lst = QAQ = 1].init();
for (i = 1; i <= n; i++) ins(s[i]); orzdalao(); Run();
return 0;
}