Java版LeetCode

Appearance

30. 串联所有单词的子串 ​

哈希表 字符串 滑动窗口

题目 ​

给定一个字符串 s和一个字符串数组 words。 words 中所有字符串 长度相同。

s中的 串联子串 是指一个包含 words 中所有字符串以任意顺序排列连接起来的子串。

  • 例如，如果 words = ["ab","cd","ef"]， 那么 "abcdef"， "abefcd"，"cdabef"， "cdefab"，"efabcd"， 和 "efcdab" 都是串联子串。 "acdbef" 不是串联子串，因为他不是任何 words 排列的连接。

返回所有串联子串在 s中的开始索引。你可以以 任意顺序 返回答案。

示例 1：

输入：s = "barfoothefoobarman", words = ["foo","bar"]
输出：[0,9]
解释：因为 words.length == 2 同时 words[i].length == 3，连接的子字符串的长度必须为 6。
子串 "barfoo" 开始位置是 0。它是 words 中以 ["bar","foo"] 顺序排列的连接。
子串 "foobar" 开始位置是 9。它是 words 中以 ["foo","bar"] 顺序排列的连接。
输出顺序无关紧要。返回 [9,0] 也是可以的。

示例 2：

输入：s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
输出：[]
解释：因为 words.length == 4 并且 words[i].length == 4，所以串联子串的长度必须为 16。
s 中没有子串长度为 16 并且等于 words 的任何顺序排列的连接。
所以我们返回一个空数组。

示例 3：

输入：s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
输出：[6,9,12]
解释：因为 words.length == 3 并且 words[i].length == 3，所以串联子串的长度必须为 9。
子串 "foobarthe" 开始位置是 6。它是 words 中以 ["foo","bar","the"] 顺序排列的连接。
子串 "barthefoo" 开始位置是 9。它是 words 中以 ["bar","the","foo"] 顺序排列的连接。
子串 "thefoobar" 开始位置是 12。它是 words 中以 ["the","foo","bar"] 顺序排列的连接。

提示：

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

题解 ​

java
public List<Integer> findSubstring(String s, String[] words) {
    List<Integer> result = new ArrayList<>();
    // 若s长度小于words总长度 则没有解
    // 单个单词长度
    int singleLength = words.length > 0 ? words[0].length() : 0;
    // 单词数组单词总长度
    int totalLength = words.length * singleLength;
    int length = s.length();
    // 数组为空或长度不满足 直接返回
    if (words.length == 0 || length < totalLength) {
        return result;
    }

    // 单词出现次数映射
    Map<String, Long> wordsCount = new HashMap<>(
        Arrays.stream(words).collect(Collectors.groupingBy(Function.identity(), Collectors.counting()))
    );

    for (int i = 0; i <= length - totalLength; i++) {
        Map<String, Long> snapshot = new HashMap<>(wordsCount);
        for (int j = 0; j < words.length; j++) {
            // 根据单词长度截取出来的单词
            String word = s.substring(i + singleLength * j, i + singleLength * (j + 1));
            // 单词匹配 计数减一并移除计数为0的单词
            if (snapshot.containsKey(word)) {
                snapshot.compute(word, (k, v) -> v - 1);
                snapshot.remove(word, 0L);
            } else {
                break;
            }
        }
        // 所有单词刚好匹配完成
        if (snapshot.isEmpty()) {
            result.add(i);
        }
    }

    return result;
}