题目描述
给定一个字符串 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
滑动窗,窗长为words所有单词的总长,将窗内字符串分为大小相等的单词,计算单词出现的次数,然后和words里单词出现的次数作比较,若相等则记录窗左边的下标。
可优化
class Solution {
public List<Integer> findSubstring(String s, String[] words) {
List<Integer> res = new ArrayList<>();
if(s==null || s.length()==0 || words==null || words.length==0) return res;
int allLength = s.length(),oneWord = words[0].length(),wordsNum = words.length;
Map<String,Integer> wordsMap = new HashMap<>();
for(int i = 0;i < wordsNum;++i){
wordsMap.put(words[i],wordsMap.getOrDefault(words[i],0)+1);
}
for(int i = 0;i < allLength-oneWord*wordsNum+1;++i){
Map<String,Integer> sMap = new HashMap<>();
String tem = s.substring(i,i+oneWord*wordsNum);
for(int j = 0;j < oneWord*wordsNum;j+=oneWord){
String temStr = tem.substring(j,j+oneWord);
sMap.put(temStr,sMap.getOrDefault(temStr,0)+1);
}
if(sMap.equals(wordsMap)) res.add(i);
}
return res;
}
}