LeetCode 第415场周赛个人题解

目录

Q1. 数字小镇中的捣蛋鬼

原题链接

思路分析

AC代码

Q2. 最高乘法得分

原题链接

思路分析

AC代码

Q3. 形成目标字符串需要的最少字符串数 I

原题链接

思路分析

AC代码

Q4. 形成目标字符串需要的最少字符串数 II

原题链接

思路分析

AC代码

---

Q1. 数字小镇中的捣蛋鬼

原题链接

Q1. 数字小镇中的捣蛋鬼

思路分析

签到题

AC代码

class Solution:
    def getSneakyNumbers(self, nums: List[int]) -> List[int]:
        cnt = Counter(nums)
        return [x for x in set(nums) if cnt[x] == 2]

Q2. 最高乘法得分

原题链接

Q2. 最高乘法得分

思路分析

划分型dp

定义状态f(i, j) 为 b 前 i 个数 和 a 前 j 个数匹配的最大得分

那么 f(i + 1, j) = max{f(i, j), f(i, j - 1) + a[j - 1] * b[j]}

时间复杂度O(4len(b))

AC代码

class Solution:
    def maxScore(self, a: list[int], b: list[int]) -> int:
        n = len(b)
        f = [[-inf] * 5 for _ in range(n + 1)]
        f[0][0] = 0
        for i in range(n):
            f[i + 1][0] = 0
            for j in range(1, 5):
                f[i + 1][j] = max(f[i + 1][j], f[i][j - 1] + a[j - 1] * b[i], f[i][j])

        return max(f[i][4] for i in range(n + 1))

Q3. 形成目标字符串需要的最少字符串数 I

原题链接

Q3. 形成目标字符串需要的最少字符串数 I

思路分析

见Q3

AC代码

auto FIO = []{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    return 0;
}();

constexpr int inf = 1e9;
struct AhoCorasick {
    static constexpr int ALPHABET = 26;
    struct Node {
        int len = 0;
        int fail = 0;
        array<int, ALPHABET> son;
        int last = 0;
        Node() : son{} {}
    };

    vector<Node> tr;

    AhoCorasick() {
        init();
    }

    void init() {
        tr.reserve(5000);
        tr.assign(2, Node());
        tr[0].son.fill(1);
        tr[1].last = 1;
    }

    int newNode() {
        tr.emplace_back();
        return tr.size() - 1;
    }

    void add(const string& s) {
        int p = 1, i = 1;
        for (char ch : s) {
            int x = ch - 'a';
            if (!tr[p].son[x])
                tr[p].son[x] = newNode();
            p = tr[p].son[x];
            tr[p].len = i ++;
        }
    }

    void build() {
        queue<int> q;
        q.push(1);
        while (q.size()) {
            int x = q.front();
            q.pop();

            for (int i = 0; i < ALPHABET; ++ i) {
                if (!tr[x].son[i]) {
                    tr[x].son[i] = tr[tr[x].fail].son[i];
                    continue;
                }
                tr[tr[x].son[i]].fail = tr[tr[x].fail].son[i];
                tr[tr[x].son[i]].last = tr[tr[tr[x].son[i]].fail].len ? 
                    tr[tr[x].son[i]].fail : tr[tr[tr[x].son[i]].fail].last;
                q.push(tr[x].son[i]);
            }
        }
    }

    int son(int p, int x) {
        return tr[p].son[x];
    }
    int fail(int p) {
        return tr[p].fail;
    }
    int last(int p) {
        return tr[p].last;
    }

    int len(int p) {
        return tr[p].len;
    }
    int size() {
        return tr.size();
    }
};

class Solution {
public:
    int minValidStrings(vector<string>& words, string target) {
        int m = words.size(), n = target.size();
        AhoCorasick ac;
        for (auto &s : words) 
            ac.add(s);
        ac.build();

        vector<int> f(n + 1, inf);
        f[0] = 0;
        int cur = 1;
        for (int i = 1; i <= n; ++ i) {
            cur = ac.son(cur, target[i - 1] - 'a');
            if (ac.len(cur))
                f[i] = min(f[i], f[i - ac.len(cur)] + 1);
        }
        return f[n] == inf ? -1 : f[n];
    }
};

Q4. 形成目标字符串需要的最少字符串数 II

原题链接

Q4. 形成目标字符串需要的最少字符串数 II

思路分析

AC自动机+dp

一个很暴力的思路就是把所有前缀都存AC自动机里面，然后在AC自动机上匹配target来进行dp

和3213. 最小代价构造字符串做法一样

但是存所有前缀会爆掉，我们修改一下AC自动机的插入，对于一个单词的插入，路径上的节点都是一个单词结尾

这样我们看似只插入了每个单词，实际上我们插入了每个单词的所有前缀

定义状态 f(i) 为 构造target 前 i 个字符所需最少字符串数

当前匹配到i，对应自动机节点为cur，那么转移方程为：

f(i) = min{f[i], f[i - ac.len(cur)] + 1}

为什么呢？不应该匹配所有前缀吗？

事实上，如果target可以构造，事实上存在合法构造，下图一定存在若干不相交的线段覆盖整个区间那么我们把target当成一个文本串，我们可以在上面匹配出所有target包含的是自动机中单词的子串，而我们每次跳fail指针，都是跳最长后缀

如果存在多组覆盖整个区间的不相交线段组，我们一定会走线段数最少的那一条

事实上，多组合法覆盖一定是相交的，我们假如正走在一条非最优解路径，到达一条线段的结束的时候我们自然会往最优解路径的方向去跳

这个过程可以手玩一下来加深理解

时间复杂度O(Σlen(words[i]))

AC代码

auto FIO = []{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    return 0;
}();

constexpr int inf = 1e9;
struct AhoCorasick {
    static constexpr int ALPHABET = 26;
    struct Node {
        int len = 0;
        int fail = 0;
        array<int, ALPHABET> son;
        int last = 0;
        Node() : son{} {}
    };

    vector<Node> tr;

    AhoCorasick() {
        init();
    }

    void init() {
        tr.reserve(5000);
        tr.assign(2, Node());
        tr[0].son.fill(1);
        tr[1].last = 1;
    }

    int newNode() {
        tr.emplace_back();
        return tr.size() - 1;
    }

    void add(const string& s) {
        int p = 1, i = 1;
        for (char ch : s) {
            int x = ch - 'a';
            if (!tr[p].son[x])
                tr[p].son[x] = newNode();
            p = tr[p].son[x];
            tr[p].len = i ++;
        }
    }

    void build() {
        queue<int> q;
        q.push(1);
        while (q.size()) {
            int x = q.front();
            q.pop();

            for (int i = 0; i < ALPHABET; ++ i) {
                if (!tr[x].son[i]) {
                    tr[x].son[i] = tr[tr[x].fail].son[i];
                    continue;
                }
                tr[tr[x].son[i]].fail = tr[tr[x].fail].son[i];
                tr[tr[x].son[i]].last = tr[tr[tr[x].son[i]].fail].len ? 
                    tr[tr[x].son[i]].fail : tr[tr[tr[x].son[i]].fail].last;
                q.push(tr[x].son[i]);
            }
        }
    }

    int son(int p, int x) {
        return tr[p].son[x];
    }
    int fail(int p) {
        return tr[p].fail;
    }
    int last(int p) {
        return tr[p].last;
    }

    int len(int p) {
        return tr[p].len;
    }
    int size() {
        return tr.size();
    }
};

class Solution {
public:
    int minValidStrings(vector<string>& words, string target) {
        int m = words.size(), n = target.size();
        AhoCorasick ac;
        for (auto &s : words) 
            ac.add(s);
        ac.build();

        vector<int> f(n + 1, inf);
        f[0] = 0;
        int cur = 1;
        for (int i = 1; i <= n; ++ i) {
            cur = ac.son(cur, target[i - 1] - 'a');
            if (ac.len(cur))
                f[i] = min(f[i], f[i - ac.len(cur)] + 1);
        }
        return f[n] == inf ? -1 : f[n];
    }
};