Leetcode 30. 与所有单词相关联的字串

最新推荐文章于 2024-03-25 19:14:51 发布

原创最新推荐文章于 2024-03-25 19:14:51 发布 · 1.8k 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一种利用KMP算法寻找一组单词在目标字符串中所有可能的组合位置的方法。通过先对单词组进行预处理和排序，减少重复计算，再利用KMP算法高效地找出每个单词在目标字符串中的匹配位置。

利用kmp记录words中每一个字符串在s中的所有匹配位置
为防止超内存，对words进行排序，并进行编号（相同的字符串为同一id），并记录每一id的出现次数
然后暴力搜索s串每一个位置是否符合要求

class Solution {
public:
    int nextval[100005];
    vector<int> A[100005];
    void get_nextval(string s) {
        int i = 0, x = -1; nextval[i] = x;
        while (i < s.size()) {
            if (x == -1 || s[i] == s[x]) nextval[++i] = ++x;
            else x = nextval[x];
        }
    }
    void strStr(string haystack, string needle, int k) {
        get_nextval(needle);
        int i = 0, j = 0;
        while (i < (int)haystack.size()) {
            if (j == -1 || haystack[i] == needle[j]) ++i, ++j;
            else j = nextval[j];
            if (j == (int)needle.size()) {
                A[i - needle.size()].push_back(k);
                j = nextval[j - 1];
                --i;
            }
        }
    }
    int L[100005], n[100005], AA[100005];
    bool _ok(int index, int cnt) {
        if (!cnt) return true;
        for (int i = 0; i < A[index].size(); ++i) {
            int x = A[index][i];
            if (AA[x]) {
                --AA[x];
                if (_ok(index + L[x], cnt - 1)) return true;
                ++AA[x];
            }
        }
        return false;
    }
    vector<int> findSubstring(string s, vector<string>& words) {
        vector<int> ans;
        sort(words.begin(), words.end());
        int h = 0, id = 0;
        if (!words.size()) return ans;
        for (int i = 0; i < words.size(); ++i) {
            h += words[i].size();
            if (i == 0) strStr(s, words[i], id);
            else if (words[i] != words[i - 1]) strStr(s, words[i], ++id);
            L[id] = words[i].size();
            ++n[id];
        }
        for (int i = 0; i + h - 1 < s.size(); ++i) {
            for (int k = 0; k <= id; ++k)
                AA[k] = n[k];
            if (_ok(i, words.size())) ans.push_back(i);
        }
        return ans;
    }
};