30. 串联所有单词的子串

2020-02-23  本文已影响0人  机器学习与自然语言处理

给定一个字符串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
上一篇 下一篇

猜你喜欢

热点阅读