给定一个字符串 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"]
输出:[]
class Solution {
void reset(int& count, int& i, unordered_map<string,int>& smap,int left){
if(!count){
i++;
}
else{
i = left +1;
}
count = 0;
smap.clear();
}
public:
vector<int> findSubstring(string s, vector<string>& words) {
int length = words.size();
if(length == 0) return vector<int>();
int wordlen = words[0].size();
unordered_map<string,int> wordmap;
for(int i=0; i<length; i++){
wordmap[words[i]]++;
}
unordered_map<string,int> smap;
vector<int> ret;
int right = 0;
int left = 0;
int count = 0;
int temp = 0;
for(int i=0; i+ wordlen <= s.size();){
string temp = s.substr(i, wordlen); //截取字符串函数必须 是 能取到最右边值的
if(wordmap.count(temp) && smap[temp]<wordmap[temp]){ //如果在words中 并且不再smap中
smap[temp]++;
if(count++ == 0){
left = i; //记录第一个能匹配上的位置
}
i+=wordlen;
if(count == length){
ret.push_back(left);
reset(count, i, smap, left);
}
}
else{
reset(count, i, smap, left);
}
}
return ret;
}
};