串联所有单词的子串

串联所有单词的子串

leetcode

给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。

注意子串要与 words 中的单词完全匹配，中间不能有其他字符，但不需要考虑 words 中单词串联的顺序。

示例 1：

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

思路和代码

用一个哈希表存储words每个单词和出现的次数，然后用 i 从头遍历s，判断从 i 开始的长度为len的单词是否是words里面的单词，同时还要是判断次数是否超过，这里同样用一个哈希来表示从 i 开始遍历到的单词的次数。

class Solution {
public:
    vector<int> ans;
    vector<int> findSubstring(string s, vector<string>& words) {
        if(s.empty()||words.empty()) 
            return ans;
        int m = s.size(), n=words.size(), length=words[0].size();
        map<string,int> m1;//存储words中单词数目
        for(auto &a:words) 
            ++m1[a];
        for(int i=0;i<=m-n*length;++i)
        {
            int j=0;
            map<string,int> m2;//存储s中已经用的单词数目
            for(;j<n;++j)
            {
                string t = s.substr(i+j*length,length);//从s中取长度为length的字符串并查                                                        //找words中是否存在
                //当前取的单词未在words中出现，不满足要求
                if(m1.find(t)==m1.end()) 
                    break;
                ++m2[t];
                if(m2[t]>m1[t])
                    break;//当前取到的子串中t的个数超过要求
            }
            //刚好取够words中单词个数
            if(j==n)
                ans.push_back(i);
        }
        return ans;
    }
};