标签:
30. Substring with Concatenation of All Words
----------------------------------------------------------------------------
Mean:
给你一个字符串s,和一个字符串集合words,找出s中words所有单词连续出现的起始下标.
analyse:
思路:窗口滑动算法.
枚举所有长度为words[0].length()的子串,然后使用窗口滑动算法来统计.
Trick:集合words是一个可重集,当连续出现的单词书大于words中的时,需要回退删除开头部分的单词,注意细节.
Time complexity: O(N)
view code
class Solution
{
public:
vector<int> findSubstring(string s, vector<string> &words)
{
vector<int> ans;
int n = s.size(), cnt = words.size();
if (n <= 0 || cnt <= 0) return ans;
// init word occurence
unordered_map<string, int> dict;
for (int i = 0; i < cnt; ++i) //统计每个单词出现的次数
dict[words[i]]++;
// travel all sub string combinations
int wl = words[0].size();
for (int i = 0; i < wl; ++i)
{
int left = i, continueWords = 0;
unordered_map<string, int> tdict;
for (int j = i; j <= n - wl; j += wl)
{
string str = s.substr(j, wl);
// a valid word, accumulate results
if (dict.count(str))
{
tdict[str]++;
if (tdict[str] <= dict[str])
continueWords++;
else
{
// a more word, advance the window left side possiablly
while (tdict[str] > dict[str])
{
string str1 = s.substr(left, wl);
tdict[str1]--;
if (tdict[str1] < dict[str1]) continueWords--;
left += wl;
}
}
// come to a result
if (continueWords == cnt)
{
ans.push_back(left);
// advance one word
tdict[s.substr(left, wl)]--;
continueWords--;
left += wl;
}
}
// not a valid word, reset all vars
else
{
tdict.clear();
continueWords = 0;
left = j + wl;
}
}
}
return ans;
}
};
LeetCode - 30. Substring with Concatenation of All Words
标签:
原文地址:http://www.cnblogs.com/crazyacking/p/5203088.html