[BZOJ3998][TJOI2015]弦论(后缀自动机+拓扑排序+贪心)

先构建出原串的后缀自动机。
然后在后缀自动机上通过拓扑排序,预处理出每一条边能到达的字符串个数(如果是位置不同则初始贡献为 Right R i g h t 集大小,本质不同则为 1 1 )。
从初始状态开始,枚举转移从a 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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值