题目:给定一个字符串 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"]
输出:[]
暴力解法
思路:
(1)用一个 unordered_map<string, int> mp储存 words 中每个词的出现次数。
(2)遍历字符串 s 的每个字符位置,外循环中每次移动1个字符,内循环每次移动一个word的长度。从该位置起不断截取 word 长度的子串,判断该 word 是否出现在 mp 中。同时通过建立一个unordered_map<string,int> tmp 保存当前单词的出现次 数。
(3)如果 mp 与 tmp 所有单词次数相同,则将当前子串起始位置存入 res 里,循环往后移动i++
class Solution {
public:
vector<int> findSubstring(string s, vector<string>& words) {
if (s.empty() || words.empty()) return {};
int n = words.size(), m = words[0].size(), j = 0;
if (s.size() < m * n) return {};
vector<int> res;
unordered_map<string, int> mp, tmp;
for (auto str : words) {
++mp[str]; //储存words每个word出现的次数
}
string str = "";
for (int i = 0; i + m * n <= s.size(); ++i) {
for (j = i; j < i + m * n; j += m) {
str = s.substr(j, m);
if (mp.find(str) == mp.end()) break;
++tmp[str]; //保存当前单词出现的次数
}
if (j == i + m * n && tmp == mp) res.push_back(i);
tmp.clear();
}
return res;
}
};
测试用例
int main(){
Solution s;
string str="barfoothefoobarman";
vector<int> res;
vector<string> words;
words.push_back("foo");
words.push_back("bar");
res=s.findSubstring(str,words);
for(vector<int>::iterator iter=res.begin();iter!=res.end();iter++){
cout<<*iter<<" ";
}
return 0;
}
结果正确,但是时间复杂度高,可以看一下别人博客的滑动窗口方法,反正我没看懂QAQ