[leetcode] 30. 串联所有单词的子串

题目

30. 串联所有单词的子串

给定一个字符串 s 和一个字符串数组 words。 words 中所有字符串 长度相同。

s 中的 串联子串 是指一个包含 words 中所有字符串以任意顺序排列连接起来的子串。

• 例如，如果 words = ["ab","cd","ef"]， 那么 "abcdef"， "abefcd"，"cdabef"， "cdefab"，"efabcd"， 和 "efcdab" 都是串联子串。 "acdbef" 不是串联子串，因为他不是任何 words 排列的连接。

返回所有串联子串在 s 中的开始索引。你可以以 任意顺序 返回答案。

示例 1：

输入：s = "barfoothefoobarman", words = ["foo","bar"]
输出：[0,9]
解释：因为 words.length == 2 同时 words[i].length == 3，连接的子字符串的长度必须为 6。
子串 "barfoo" 开始位置是 0。它是 words 中以 ["bar","foo"] 顺序排列的连接。
子串 "foobar" 开始位置是 9。它是 words 中以 ["foo","bar"] 顺序排列的连接。
输出顺序无关紧要。返回 [9,0] 也是可以的。

示例 2：

输入：s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
输出：[]
解释：因为 words.length == 4 并且 words[i].length == 4，所以串联子串的长度必须为 16。
s 中没有子串长度为 16 并且等于 words 的任何顺序排列的连接。
所以我们返回一个空数组。

示例 3：

输入：s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
输出：[6,9,12]
解释：因为 words.length == 3 并且 words[i].length == 3，所以串联子串的长度必须为 9。
子串 "foobarthe" 开始位置是 6。它是 words 中以 ["foo","bar","the"] 顺序排列的连接。
子串 "barthefoo" 开始位置是 9。它是 words 中以 ["bar","the","foo"] 顺序排列的连接。
子串 "thefoobar" 开始位置是 12。它是 words 中以 ["the","foo","bar"] 顺序排列的连接。

提示：

• 1 <= s.length <= 104
• 1 <= words.length <= 5000
• 1 <= words[i].length <= 30
• words[i] 和 s 由小写英文字母组成

代码

#include<bits/stdc++.h>
using namespace std;

class Solution {
public:
    vector<int> ans;
    map<string,int> words2idx;

    vector<int> findSubstring(string s, vector<string>& words) {
        for (int i = 0; i < words.size(); i++) words2idx.insert(pair<string, int>(words[i], i));
        int word_len = words[0].length(), seq_len = words.size();
        
        vector<vector<int>> dp(word_len + 10, vector<int>(seq_len + 10, 0));
        for (int i = 0; i < word_len; i++){
            for (int j = 0; j < seq_len; j++){
                dp[i][words2idx[words[j]]]++;
            }
        }

        vector<string> subs(word_len + 10);
        vector<int> cnt(word_len + 10, seq_len);

        for (int i = 0; i < s.length(); i++){
            for (int j = 0; j < word_len; j++){
                if (i + j >= s.length()) continue;
                subs[j] += s[i + j];
                if (subs[j].length() % word_len == 0){
                    auto iter = words2idx.find(subs[j].substr(subs[j].length() - word_len));
                    if (iter != words2idx.end()){
                        if (dp[j][iter->second] > 0) cnt[j]-=1;
                        else cnt[j] -= 1e4;
                        dp[j][iter->second]--;
                    }
                }

                if (subs[j].length() == word_len * seq_len){
                    if (cnt[j] == 0){
                        ans.push_back(i + j - word_len * seq_len + 1);
                    }
                    string tmp = subs[j].substr(0, word_len);
                    auto iter = words2idx.find(tmp);
                    if (iter != words2idx.end()){
                        if (dp[j][iter->second] >= 0) cnt[j]+=1;
                        else cnt[j] += 1e4;
                        dp[j][iter->second]++;
                    }
                    subs[j].erase(0, word_len);
                }
            }
        }
        return ans;
    }
};

int main(){
    int n;
    string s;
    cin>>n;
    cin>>s;
    vector<string> words(n);
    for (int i = 0; i < n; i++) cin>>words[i];
    vector<int> ans = Solution().findSubstring(s, words);
    for (int i = 0; i < ans.size(); i++) cout<<ans[i]<<" ";
}

滑动窗口。

其实可以不用开 words[i].length 个数组，可以选择不同的起点，然后一次在末尾增加 words[i].length 个字符，在头部删去 words[i].length 字符。相对而言更好理解。