题目描述
给定一个字符串 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"]
输出:[]
解答
不知道问题出在哪里,本地测试的时候可以通过leetcode上没有通过的测试用例。但是在leetcode上,提交出错。
思路:
将words中所有元素用map保存,记录每个元素的个数。然后将字符串s按照所有words中元素的总长来截取。判读每一个截取的字符串是否包含words中所有元素。如果包含那么就返回true。保存当前元素索引位置。
class Solution {
public:
vector<int> findSubstring(string s, vector<string>& words) {
vector<int> res;
map<string,int> dict;
int len;
for(int i=0;i<words.size();++i)
{
dict[words[i]]++;
len+=words[i].size();
}
if(len>s.size()) return res;
int word_len=words[0].size();
for(int i=0;i<=s.size()-len;++i)
{
string temp=s.substr(i,len);
if(fun(temp,dict,word_len)) res.push_back(i);
}
return res;
}
bool fun(string &temp,map<string,int> dict,int word_len)
{
for(int i=0;i<temp.size();)
{
string sub=temp.substr(i,word_len);
if(dict.find(sub)==dict.end()) return false;
else
{
dict[sub]--;
if(dict[sub]==0) dict.erase(sub);
}
i+=word_len;
}
if(dict.empty()) return true;
else
return false;
}
};