2020 CCPC Wannafly Winter Camp Day2-K-破忒头的匿名信

时间:2023-11-15 11:47:38

题目传送门

sol:先通过AC自动机构建字典,用$dp[i]$表示长串前$i$位的最小代价,若有一个单词$s$是长串的前$i$项的后缀,那么可以用$dp[i - len(s)] + val(s)$转移到$dp[i]$。

  • AC自动机
    #include <bits/stdc++.h>
    using namespace std;
    typedef long long LL;
    typedef pair<int, int> PII;
    const int MAXN = ;
    const LL INF = 0x3f3f3f3f3f3f3f3f;
    struct Trie {
    int son[MAXN][], fail[MAXN], last[MAXN], dep[MAXN];
    LL val[MAXN], dp[MAXN]; int tot, root;
    int add_node(int d) {
    memset(son[tot], -, sizeof(son[tot]));
    dep[tot] = d; val[tot] = INF;
    return tot ++;
    }
    void init() {
    memset(dp, INF, sizeof(dp));
    dp[] = tot = ;
    root = add_node();
    }
    void insert(char* s, LL v) {
    int p = root;
    for (int i = ; s[i]; i++) {
    int index = s[i] - 'a';
    if (son[p][index] == -)
    son[p][index] = add_node(i + );
    p = son[p][index];
    }
    val[p] = min(val[p], v);
    }
    void build() {
    queue<int> que; fail[root] = root;
    for (int i = ; i < ; i++) {
    if (son[root][i] == -) son[root][i] = root;
    else {
    fail[son[root][i]] = root;
    last[son[root][i]] = root;
    que.push(son[root][i]);
    }
    }
    while (!que.empty()) {
    int p = que.front(); que.pop();
    for (int i = ; i < ; i++) {
    if (son[p][i] == -) son[p][i] = son[fail[p]][i];
    else {
    fail[son[p][i]] = son[fail[p]][i];
    if (val[son[fail[p]][i]] != INF)
    last[son[p][i]] = son[fail[p]][i];
    else
    last[son[p][i]] = son[last[p]][i];
    que.push(son[p][i]);
    }
    }
    }
    }
    LL slove(char* s) {
    int p = root, len = strlen(s);
    for (int i = ; i < len; i++) {
    int index = s[i] - 'a';
    p = son[p][index];
    for (int tmp = p; tmp != root; tmp = last[tmp])
    dp[i + ] = min(dp[i + ], dp[i + - dep[tmp]] + val[tmp]);
    }
    if (dp[len] == INF) return -;
    else return dp[len];
    }
    } ac;
    char s[MAXN];
    int main() {
    int n, v;
    scanf("%d", &n); ac.init();
    for (int i = ; i <= n; i++) {
    scanf("%s%d", s, &v);
    ac.insert(s, v);
    }
    ac.build();
    scanf("%s", s);
    printf("%lld\n", ac.slove(s));
    return ;
    }

    AC自动机不熟练,硬怼了三天才敲出来,一直超时。又学到了两个AC自动机的优化。