给定一个字符串 s s s 和一些长度相同的单词 w o r d s words words。找出 s s s 中恰好可以由 w o r d s words words 中所有单词串联形成的子串的起始位置。
注意子串要与 w o r d s words words 中的单词完全匹配,中间不能有其他字符,但不需要考虑 w o r d s words words 中单词串联的顺序。
示例 1:
输入:
s = "barfoothefoobarman",
words = ["foo","bar"]
输出:[0,9]
解释:
从索引 0 和 9 开始的子串分别是 "barfoor" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。
示例 2:
输入:
s = "wordgoodgoodgoodbestword",
words = ["word","good","best","word"]
输出:[]
思路:由于 w o r d s words words中每个单词的长度都相同,所以可以计算出 w o r d s words words的长度 n ∗ w o r d s [ 0 ] . s i z e ( ) n*words[0].size() n∗words[0].size(), n n n表示 w o r d s words words中单词的个数;用 m a p map map存放 w o r d s words words中单词出现的次数令为 m p 1 mp1 mp1, k e y key key为单词, v a l u e value value为单词出现的次数;然后遍历 s s s字符串,每次取 s s s中长度为 w o r d s [ 0 ] . s i z e ( ) words[0].size() words[0].size()的字符串在 m p 1 mp1 mp1中查找,若未找到,则说明从当前i开始的字符串不满足要求,接着遍历,当取到 n n n个字符串并且都在 m p 1 mp1 mp1中查找到则说明该子串符合要求
class Solution {
public:
vector<int> ans;
vector<int> findSubstring(string s, vector<string>& words) {
if(s.empty()||words.empty())
return ans;
int m = s.size(), n=words.size(), length=words[0].size();
map<string,int> m1;//存储words中单词数目
for(auto &a:words)
++m1[a];
for(int i=0;i<=m-n*length;++i)
{
int j=0;
map<string,int> m2;//存储s中已经用的单词数目
for(;j<n;++j)
{
string t = s.substr(i+j*length,length);//从s中取长度为length的字符串并查找words中是否存在
//当前取的单词未在words中出现,不满足要求
if(m1.find(t)==m1.end())
break;
++m2[t];
if(m2[t]>m1[t])
break;//当前取到的子串中t的个数超过要求
}
//刚好取够words中单词个数
if(j==n)
ans.push_back(i);
}
return ans;
}
};