题目:
给定一个字符串 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
1.统计words中每个单词出现次数记录在字典中。
2.遍历s,寻找是否有和字典中相同的单词。若有把字典中该单词的索引减1,代表之后不再重复验证,题目要求words所有单词组成的字符串用了匹配。
3.若最后字典中单词匹配完。记录初始匹配的位置。
代码:
def findSubstring(self, s: str, words: List[str]) -> List[int]:
if not words:
return []
ans=[]
slen=len(words[0])
idx=0
worddic={}
for word in words:
if word in worddic:
worddic[word]+=1
else:
worddic[word]=1
for i in range(len(s)-slen*len(words)+1):
count=0
idx=i
wordcopy=worddic.copy()
while count<len(words):
cur=s[idx:idx+slen]
strr=cur
if strr in wordcopy and wordcopy[strr]!=0:
idx+=slen
count+=1
wordcopy[strr]-=1
else:
break
if count==len(words):
ans.append(i)
return ans