Leetcode 30. 串联所有单词的子串 C++

Leetcode 30. 串联所有单词的子串

题目

给定一个字符串 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"]
输出：[]

题解

通过阅读题目可以知道，words中的每个单词都是一样长的，因此我们不妨用哈希表统计一下words中每个单词出现的次数。假设words中的每个单词的长度为len，遍历我们的下标，对于每个下标，我们用len来截取字符串，如果当前截取的字符串未在words中出现或者出现次数多了，则结束循环。若n*len均通过检测，则此时的下标一定是符合题意得一个下标。

代码

vector<int> findSubstring(string s, vector<string>& words) {
        if(words.empty() || s.empty())  return {};
        vector<int> ans;
        int len = words[0].length(),n=words.size(),total=n*len;;
        int l = s.length();
        unordered_map<string,int> list;
        int i,j,left;
        for(i=0; i<n; i++)  list[words[i]]++;
        for(i=0; i<=l-total; i++){
            unordered_map<string,int> window;
            bool flag = true;
            left = i;
            string str;
            while(left-i<total){
                str = s.substr(left,len);
                if(list.count(str)==1 && window[str]!=list[str]){
                    window[str]++;
                    left += len;
                }else{
                    flag = false;
                    break;
                }   
            }
            if(flag)  ans.push_back(i);
        }
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/substring-with-concatenation-of-all-words
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。