1. 程式人生 > 其它 >LeetCode-30 串聯所有單詞的子串

LeetCode-30 串聯所有單詞的子串

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/substring-with-concatenation-of-all-words

題目描述

給定一個字串 s 和一些 長度相同 的單詞 words 。找出 s 中恰好可以由 words 中所有單詞串聯形成的子串的起始位置。

注意子串要與 words 中的單詞完全匹配,中間不能有其他字元 ,但不需要考慮 words 中單詞串聯的順序。

 

示例 1:

輸入:s = "barfoothefoobarman", words = ["foo","bar"]
輸出:[0,9]
解釋:
從索引 0 和 9 開始的子串分別是 "barfoo" 和 "foobar" 。
輸出的順序不重要, [9,0] 也是有效答案。
示例 2:

輸入:s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
輸出:[]
示例 3:

輸入:s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
輸出:[6,9,12]
 

提示:

1 <= s.length <= 104
s 由小寫英文字母組成
1 <= words.length <= 5000
1 <= words[i].length <= 30
words[i] 由小寫英文字母組成

解題思路

題目中重點強調了長度相同,這就是解題的關鍵,由於長度相同,我們完全可以知道子串的長度就是單個單詞長度乘以單詞個數,所以可以使用一個子串長度的滑動視窗,依次判斷各個起始位置的滑動視窗內子串是否符合題目要求。

同樣由於單詞長度固定已知,可以直接將視窗內的子串拆解成一個一個單詞判斷是否在words陣列內,為了提高速度,可以把words陣列內的單詞存入一個雜湊表。

程式碼展示

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        vector<int> viRet;
        unordered_map<string, int> striMap, strimapTemp;
        for(auto word: words)
        {
            striMap[word]
++; } int iLen = words.size() * words[0].size(); for(int i = 0; i < s.size() - iLen + 1; i++) { strimapTemp = striMap; bool bTemp = true; for(int j = 0; j < words.size(); j++) { string word = s.substr(i + j * words[0].size(), words[0].size()); if(strimapTemp[word] == 0) { bTemp = false; break; } strimapTemp[word]--; } if(bTemp) viRet.push_back(i); } return viRet; } };

 

執行結果