给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。
注意子串要与 words 中的单词完全匹配,中间不能有其他字符,但不需要考虑 words 中单词串联的顺序。
示例 1:输入:
s = "barfoothefoobarman",
words = ["foo","bar"]
输出:[0,9]
解释:
从索引 0 和 9 开始的子串分别是 "barfoor" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。
示例 2:输入:
s = "wordgoodgoodgoodbestword",
words = ["word","good","best","word"]
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/substring-with-concatenation-of-all-words
思路:建立两个哈希表,分别将字符串和字符列表存入,对比对应单词的出现次数。
class Solution:
def findSubstring(self, s: str, words: List[str]) -> List[int]:
if len(s) == 0 or len(words) == 0:
return []
hashmap1 = {}
wordnum = len(words)
wordlenth = len(words[0])
ans = []
for i in range(0,len(words)):
if hashmap1.get(words[i]) is None:
hashmap1[words[i]]=1
else:
hashmap1[words[i]]+=1
for i in range(0,len(s)-wordlenth*wordnum+1):
hashmap2={}
sort=1
for j in range(0,len(words)):
key=s[i+j*wordlenth:i+(j+1)*wordlenth]
if(hashmap1.get(key) is None):
sort=0
break
if(hashmap2.get(key) is None):
hashmap2[key]=1
else:
hashmap2[key]+=1
if(hashmap2[key]>hashmap1[key]):
sort=0
break
if(sort==1):
ans.append(i)
return ans