题目:
给定一个字符串 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 <= 104
- s 由小写英文字母组成
- 1 <= words.length <= 5000
- 1 <= words[i].length <= 30
- words[i] 由小写英文字母组成
class Solution {
public List<Integer> findSubstring(String s, String[] words) {
List<Integer> list = new ArrayList<>();
// 排除特殊情况
if(s==null || words==null || words.length==0 || words[0]==null
|| s.length()==0 || words[0].length() ==0 || s.length()<words.length*words[0].length()){
return list;
}
// <单词,出现次数>
Map<String, Integer> map = new HashMap<>();
for (String word : words) {
map.put(word,map.getOrDefault(word,0)+1);
}
int len = words[0].length(); // 单词长度
int strLen = s.length(); // 字符串长度
int lastStart = len * words.length - len; // 下一个单词的起始位置
for (int i = 0; i < len; i++) {
for (int j = i; j <= strLen - len - lastStart; j+=len) {
String str = s.substring(j,j+len);
if(map.containsKey(str)){
Map<String, Integer> map1 = new HashMap<>();
// 从后往前依次对比
int index = j + lastStart;
String str1 = s.substring(index,index+len);
while(index>=j && map.containsKey(str1)){
if(map1.getOrDefault(str1,0)<map.get(str1)){
map1.put(str1,map1.getOrDefault(str1,0)+1);
}else {
break;
}
index = index - len;
if(index<j) {
break;
}
str1 = s.substring(index,index+len);
}
// 完全匹配所有单词
if(j>index){
list.add(j);
}else { // 从index 到 index+len 这个单词不能匹配
j = index;
}
}
}
}
return list;
}
}