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; } };
執行結果