给定一个字符串s 和一些长度相同的单词* words*。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。
注意子串要与 words 中的单词完全匹配,中间不能有其他字符,但不需要考虑 words 中单词串联的顺序。
来源:力扣(LeetCode)
示例:
输入:
s = "barfoothefoobarman",
words = ["foo","bar"]
输出:[0,9]
解释:
从索引 0 和 9 开始的子串分别是 "barfoo" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。
链接:https://leetcode-cn.com/problems/substring-with-concatenation-of-all-words
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:一道典型的滑动窗口问题,用一个字符串存储一个和所有单词长度总和相等的序列,在用一个字典记录不同word出现的次数是否和目标words中的元素是否相等。如果相同就添加到结果中。
class Solution:
def findSubstring(self, s: str, words: List[str]) -> List[int]:
if not s or not words: return []
from collections import Counter
n = len(s)
res = []
one_word = len(words[0])
all_words = len(words) * one_word
word_dict = Counter(words)
for i in range(0, n-all_words+1):
tmp = s[i:i+all_words]
c_tmp = []
for j in range(0, len(tmp), one_word):
c_tmp.append(tmp[j:j+one_word])
if Counter(c_tmp) == word_dict:
res.append(i)
return res