LeetCode-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"]
输出：[]
示例 3：

输入：s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
输出：[6,9,12]
 

提示：

1 <= s.length <= 10⁴
s 由小写英文字母组成
1 <= words.length <= 5000
1 <= words[i].length <= 30
words[i] 由小写英文字母组成

解题思路

题目中重点强调了长度相同，这就是解题的关键，由于长度相同，我们完全可以知道子串的长度就是单个单词长度乘以单词个数，所以可以使用一个子串长度的滑动窗口，依次判断各个起始位置的滑动窗口内子串是否符合题目要求。

同样由于单词长度固定已知，可以直接将窗口内的子串拆解成一个一个单词判断是否在words数组内，为了提高速度，可以把words数组内的单词存入一个哈希表。

代码展示

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        vector<int> viRet;
        unordered_map<string, int> striMap, strimapTemp;
        for(auto word: words)
        {
            striMap[word]++;
        }
        int iLen = words.size() * words[0].size();
        for(int i = 0; i < s.size() - iLen + 1; i++)
        {
            strimapTemp = striMap;
            bool bTemp = true;
            for(int j = 0; j < words.size(); j++)
            {
                string word = s.substr(i + j * words[0].size(), words[0].size());
                if(strimapTemp[word] == 0)
                {
                    bTemp = false;
                    break;
                }
                strimapTemp[word]--;
            }
            if(bTemp)
                viRet.push_back(i);
        }
        return viRet;
    }
};

 

运行结果