题目:
给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。
注意子串要与 words 中的单词完全匹配,中间不能有其他字符,但不需要考虑 words 中单词串联的顺序。
题目链接:Substring with Concatenation of All Words
C++:
class Solution {
public:
vector<int> findSubstring(string s, vector<string>& words) {
multiset<string> wordSet;
vector<int> ret;
int cnt = words.size();
if(cnt == 0 || s.size() == 0)
return ret;
int aWordLen = words[0].size();
for(auto itr : words){
wordSet.insert(itr);
}
int len = cnt * aWordLen;
for(int i = 0; i < s.size() - len + 1; i++){
string tmp = s.substr(i, len);
auto j = wordSet.begin();
while(j != wordSet.end()){
int flag = tmp.find(*j);
while(flag % aWordLen != 0 && flag != tmp.npos){
flag = tmp.find(*j, flag + 1);
}
if(flag != tmp.npos)
tmp.erase(flag, aWordLen);
else
break;
j++;
}
if(tmp.empty())
ret.push_back(i);
}
return ret;
}
};