题目
给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。
注意子串要与 words 中的单词完全匹配,中间不能有其他字符,但不需要考虑 words 中单词串联的顺序。
示例 1:
输入:
s = "barfoothefoobarman",
words = ["foo","bar"]
输出:[0,9]
解释:
从索引 0 和 9 开始的子串分别是 "barfoor" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。
示例 2:
输入:
s = "wordgoodgoodgoodbestword",
words = ["word","good","best","word"]
输出:[]
思路
这道题得抓住红色的关键字,假设 words 数组中有n个单词,每个单词的长度均为 len,那么实际上这道题就让我们求出所有长度为 n*len 的子串,使得其刚好是由 words 数组中的所有单词组成。
那么我们就需要经常判断s串中长度为 len 的子串是否是 words 中的单词,为了快速的判断,就要使用哈希表,同时由于 words 数组可能有重复单词,所以就要用 HashMap 来建立所有的单词和其出现次数之间的映射,即统计每个单词出现的次数。
class Solution {
public:
vector<int> findSubstring(string s, vector<string>& words) {
vector<int> res;
if(s.empty() || words.empty()) return res;
int n = int(s.size()),window = int(words[0].size()),num = int(words.size());
if(n < window*num) return res;
// 使用字典记录words
unordered_map<string,int> m;
for(auto word:words) m[word]++;
// 需要遍历s中所有长度为 num*window 的子串,当剩余子串的长度小于 num*window 时,就不用再判断了
for(int i = 0;i <= n-window*num;i++){
int cnt = 0;
// 新建一个字典,去记录每个词使用次数
unordered_map<string,int> cur_m;
for(int j = i;cnt < num;j+=window){
string str = s.substr(j,window);
// 在原字典找不到,或者使用超过次数,说明不匹配
if(m.count(str) and m[str] > cur_m[str]) {cur_m[str]++;cnt++;}
else break;
}
if(cnt == num) res.push_back(i);
}
return res;
}
};