题目描述
给定一个字符串 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”]
输出: []
题目分析
首先用map记录每个word单词出现的次数。接着再用一个map记录words长度的字符串中找到了几个word,并且记录word出现的次数。
时间复杂度是O(M*N),M是word的个数,N是字符串的长度。
代码
Java代码如下:
class Solution {
public List<Integer> findSubstring(String s, String[] words) {
List<Integer> result = new ArrayList<>();
if(words.length==0)
{
return result;
}
Map<String,Integer> map = new HashMap<>();
for(String word:words)
{
map.put(word,map.getOrDefault(word,0)+1);
}
int n=words.length;
int len=words[0].length();
Map<String,Integer> found = new HashMap<>();
for(int i=0;i<=s.length()-n*len;i++)
{
found.clear();
int j=0;
for(j=0;j<n;j++)
{
String str =s.substring(i+j*len,i+j*len+len);
if(map.containsKey(str))
{
found.put(str,found.getOrDefault(str,0)+1);
if(found.get(str)>map.get(str))
{
break;
}
} else
{
break;
}
}
if(j==n)
{
result.add(i);
}
}
return result;
}
}