30. 串联所有单词的子串【leetcode】

原题地址：https://leetcode-cn.com/problems/substring-with-concatenation-of-all-words/

好久没刷题了。今天再刷题记录下：

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

我的思路：

用map先把words中的单词和出现次数保存下来，再用两个指针l,r，[l,r)代表的就是符合要求的子串。

每次循环看subs= substr(r,len)在map中是否为0，可分为以下几种情况：

            if (mapWords[subs] != 0) //下面这个单词可以加入子串
            {
                r = r + wlen; 
                --mapWords[subs];
            }
            else
            {
                if (l == r)   //l和r相等，同时前进一步
                {
                    ++l;
                    ++r;                    
                }
                else   //l和r不相等，要更新map，并且让l向前一步，r回来
                {
                    mapWords=mapWords2;
                    ++l;
                    r = l;
                }
            }

完整代码：判断子串是否已经符合条件那里写的不太好。

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        map<string, int> mapWords;
        map<string, int> mapWords2;
        vector<int> res;
        if (s.size() < words.size() || words.size() == 0)
        {
            return res;
        }
        int wlen = words[0].size();
        for (int i = 0 ; i < words.size() ; ++i)
        {
            ++mapWords[words[i]];
        }
        mapWords2=mapWords;
        int l = 0, r = 0 ,sign = 0;
        while(r <= s.size())
        {
            for (auto i = mapWords.begin() ; i != mapWords.end() ; ++i)
            {
                if (i->second != 0)
                {
                    sign = 1;
                    break;
                }
            }
            if (sign == 0)
            {
                res.push_back(l);
            }
            else
            {
                sign = 0;
            }
            string subs = s.substr(r, wlen);
            if (mapWords[subs] != 0)
            {
                r = r + wlen; 
                --mapWords[subs];
            }
            else
            {
                if (l == r)
                {
                    ++l;
                    ++r;                    
                }
                else
                {
                    mapWords=mapWords2;
                    ++l;
                    r = l;
                }
            }
        }
        return res;
    }
};