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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/substring-with-concatenation-of-all-words
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
    public List<Integer> findSubstring(String s, String[] words) {
        List<Integer> res = new ArrayList<>();
        if(s == null || s.equals("") || words == null || words.length == 0) {
            return res;
        }
        Map<String, Integer> map = new HashMap<>();
        for(String word: words) {
            map.put(word, map.getOrDefault(word, 0) + 1);
        }
        int wordLen = words[0].length(), stringLen = words.length * wordLen;
        for(int i = 0; i < wordLen; i++) {
            Map<String, Integer> tempMap = new HashMap<>();
            // 匹配单词数目count
            int left = i, right = i, count = 0;
            // 滑动窗口
            while(right + wordLen <= s.length()) {
                String w = s.substring(right, right + wordLen);
                tempMap.put(w, tempMap.getOrDefault(w, 0) + 1);
                right += wordLen;
                count++;
                while(tempMap.getOrDefault(w, 0) > map.getOrDefault(w, 0)) {
                    String tw = s.substring(left, left + wordLen);
                    tempMap.put(tw, tempMap.get(tw) - 1);
                    left += wordLen;
                    count--;
                }
                if(count == words.length) {
                    res.add(left);
                }
            }
        }
        return res;
    }
}