Leetcode30 串联所有单词的子串 C++

题目描述

给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。
注意子串要与 words 中的单词完全匹配，中间不能有其他字符，但不需要考虑 words 中单词串联的顺序。
示例 1：

输入：
  s = "barfoothefoobarman",
  words = ["foo","bar"]
输出：[0,9]
解释：
从索引 0 和 9 开始的子串分别是 "barfoor" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。

示例 2：

输入：
  s = "wordgoodgoodgoodbestword",
  words = ["word","good","best","word"]
输出：[]

解答

不知道问题出在哪里，本地测试的时候可以通过leetcode上没有通过的测试用例。但是在leetcode上，提交出错。
思路：
将words中所有元素用map保存，记录每个元素的个数。然后将字符串s按照所有words中元素的总长来截取。判读每一个截取的字符串是否包含words中所有元素。如果包含那么就返回true。保存当前元素索引位置。

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        vector<int> res;
        map<string,int> dict;
        int len;
        for(int i=0;i<words.size();++i)
        {
            dict[words[i]]++;
            len+=words[i].size();
        }
        if(len>s.size()) return res;
        int word_len=words[0].size();
        for(int i=0;i<=s.size()-len;++i)
        {
            string temp=s.substr(i,len);
            if(fun(temp,dict,word_len)) res.push_back(i);
        }
        return res;
    }
    
    bool fun(string &temp,map<string,int> dict,int word_len)
    {
        for(int i=0;i<temp.size();)
        {
            string sub=temp.substr(i,word_len);
            if(dict.find(sub)==dict.end()) return false;
            else
            {
                dict[sub]--;
                if(dict[sub]==0) dict.erase(sub);
            }
            i+=word_len;
        }
        if(dict.empty()) return true;
        else 
            return false;
    }
};