题目描述:
给定一个字符串 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:
主要思路:
(1)先使用unordered_map统计给出的字典中各个单词的个数;
(2)然后从头开始,对每个位置判断从该位置作为起点,能否存在一个全部由字典中单词组成的子字符串;
(3)并将各个满足要求的位置压入到结果中;
class Solution {
public:
vector<int> findSubstring(string s, vector<string>& words) {
if(s.size()<words.size()*words[0].size()){//处理特殊的情形
return {};
}
//统计字典中各个单词出现的次数
unordered_map<string,int> mp;
for(string& word:words){
++mp[word];
}
//存储结果,既各个可能的起始位置
vector<int> res;
int step=words[0].size();//单词的长度,既每次需要跳转的长度
//终止的位置,就该位置之后,长度就不满足要求了,没必要再进行判断
int end=s.size()-step*words.size()+1;
//判断各个可能的位置
for(int i=0;i<end;++i){
//若起始位置的单词满足要求,则开始判断
if(mp.count(s.substr(i,step))){
//辅助的统计映射数据
unordered_map<string,int> tmp=mp;
int j=i;
//直到字典中的所有的单词都出现过,则跳出循环
while(!tmp.empty()){
string str=s.substr(j,step);//当前位置的对应的单词
//若当前位置对应的单词存在字典中
if(tmp.count(str)){
--tmp[str];//将出现过的频次减1
if(tmp[str]==0){//若减到了0,则删除当前统计的单词
tmp.erase(str);
}
}
else{//若当前单词不存在字典中,则直接跳出循环
break;
}
//判断下一个位置
j+=step;
}
//若全部的单词都被删除了,则说明当前位置对应的子串满足要求,则将当前位置压入到结果中
if(tmp.empty()){
res.push_back(i);
}
}
}
//返回结果
return res;
}
};