题目描述:
给定一个字符串 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]
思路解析:
- map中保存给定words数组中每个单词的数目(word,数量)
- 计算数组中每个单词的长度onelen,和数组的总长度len
- 从i开始,左右指针,右指针每次移动onelen个长度。i的取值范围是onelen的长度。因为所有的word长度是一样的,起始指针的范围就是word的长度,while循环会保证所有的长度都被覆盖。因为在for循环最后一次的时候,s中最后刚好不符合要求的长度是两个字符。
- 【right,right+onelen】得到一个新的word1
- 如果s==map中没有word1,匹配失败,right移动。
- 如果s==map中有word1,submap中统计截止当前子串中word1出现的次数(是当前的子串),并count统计当前窗口字符数。
如果当前子串submap中次数大于给定数组中该word出现的次数,则左指针左移,并从submap中删除【左,左+onelen】的单词的次数,直到submap中次数不大于给定数组中该word出现的次数
如果不大于,res中记录left的值 - i移动,更新子串起点
right范围是onelen的长度,每次看右边的字符串是不是在words中,并且当前子串的word数量不大于给定数组。
代码:
public List<Integer> findSubstring(String s, String[] words) {
List<Integer> res = new ArrayList<>();
// words与数量对应的wordsmap
Map<String, Integer> wordsMap = new HashMap<>();
if (s.length() == 0 || words.length == 0) return res;
for (String word: words) {
if (s.indexOf(word) < 0) return res;
wordsMap.put(word, wordsMap.getOrDefault(word, 0) + 1);
}
// 每个单词的长度和总长度
int oneLen = words[0].length(), wordsLen = oneLen * words.length;
if (wordsLen > s.length()) return res;
// for循环的长度是onelen的长度
for (int i = 0; i < oneLen; ++i) {
int left = i, right = i, count = 0;
// 统计当前的子串
Map<String, Integer> subMap = new HashMap<>();
// 右窗口不能超出主串长度
while (right + oneLen <= s.length()) {
// 得到一个单词
String word = s.substring(right, right + oneLen);
// 有窗口右移
right += oneLen;
// words[]中没有这个单词,那么当前窗口肯定匹配失败,直接右移到这个单词后面
if (!wordsMap.containsKey(word)) {
left = right;
// 窗口内单词统计map清空,重新统计
subMap.clear();
// 符合要求的单词数清0
count = 0;
} else {
// 统计当前子串中这个单词出现的次数
subMap.put(word, subMap.getOrDefault(word, 0) + 1);
++count;
// 如果这个单词出现的次数大于words[]中它对应的次数,又由于每次匹配和words长度相等的子串
// 如 ["foo","bar","foo","the"] "| foobarfoobar| foothe"
// 第二个bar虽然是words[]中的单词,但是次数抄了,那么右移一个单词长度后 "|barfoobarfoo|the"
// bar还是不符合,所以直接从这个不符合的bar之后开始匹配,也就是将这个不符合的bar和它之前的单词(串)全移出去
// 如果子串中word次数大于word次数,则不符合,则移动左指针,直到符合
while (subMap.getOrDefault(word, 0) > wordsMap.getOrDefault(word, 0)) {
// 从当前窗口字符统计map中删除从左窗口开始到数量超限的所有单词(次数减一)
String w = s.substring(left, left + oneLen);
subMap.put(w, subMap.getOrDefault(w, 0) - 1);
// 符合的单词数减一
--count;
// 左窗口位置右移
left += oneLen;
}
// 当前窗口字符串满足要求
if (count == words.length) res.add(left);
}
}
}
return res;
}