leetcode30 串联所有单词的子串（hard）

题目

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

 

思路

这道题得抓住红色的关键字，假设 words 数组中有n个单词，每个单词的长度均为 len，那么实际上这道题就让我们求出所有长度为 n*len 的子串，使得其刚好是由 words 数组中的所有单词组成。

那么我们就需要经常判断s串中长度为 len 的子串是否是 words 中的单词，为了快速的判断，就要使用哈希表，同时由于 words 数组可能有重复单词，所以就要用 HashMap 来建立所有的单词和其出现次数之间的映射，即统计每个单词出现的次数。

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        vector<int> res;
        if(s.empty() || words.empty()) return res;
        int n = int(s.size()),window = int(words[0].size()),num = int(words.size());
        if(n < window*num) return res;
        // 使用字典记录words
        unordered_map<string,int> m;
        for(auto word:words) m[word]++;
        // 需要遍历s中所有长度为 num*window 的子串，当剩余子串的长度小于 num*window 时，就不用再判断了
        for(int i = 0;i <= n-window*num;i++){
            int cnt = 0;
            // 新建一个字典，去记录每个词使用次数
            unordered_map<string,int> cur_m;
            for(int j = i;cnt < num;j+=window){
                string str = s.substr(j,window);
                // 在原字典找不到，或者使用超过次数，说明不匹配
                if(m.count(str) and m[str] > cur_m[str]) {cur_m[str]++;cnt++;}
                else break;
            }
            if(cnt == num) res.push_back(i);
        }
        return res;
    }
};