把握串联所有单词的精髓：巧用滑动窗口算法

前端

2023-11-28 14:55:43

前言

欢迎来到力扣刷题记的第 30 篇，今天我们共同探秘「串联所有单词的子串」这一算法难题。在这个过程中，我们将深入浅出地剖析滑动窗口算法的精妙之处，帮助你解锁算法世界的新篇章。

题目

给定一个字符串 s 和一些长度相同的单词 words，找出 s 中恰好可以由 words 中所有单词串联形成的子串。返回所有这样的子串的起始索引。

滑动窗口算法的闪耀登场

解决该难题的关键在于滑动窗口算法的巧妙应用。该算法利用了一个不断移动的窗口来扫描字符串，在窗口内进行特定的操作。

算法步骤

初始化窗口： 以单词 words 中第一个单词的长度初始化窗口，将其放置在字符串 s 的开头。
检查窗口： 检查窗口内的字符是否与 words 中当前单词匹配。如果是，则继续下一步；否则，移动窗口。
匹配成功： 如果匹配成功，则将 words 中的当前单词从窗口中移除，并将其与 words 中下一个单词进行比较。
匹配失败： 如果匹配失败，则将窗口向右移动一个字符。
重复以上步骤： 重复步骤 2-4，直至遍历完整个字符串 s。

代码示例

def findSubstring(s, words):
    word_len = len(words[0])
    words_set = set(words)
    result = []

    for i in range(len(s) - word_len * len(words) + 1):
        window = s[i:i + word_len * len(words)]
        matched_words = []

        for j in range(0, len(window), word_len):
            word = window[j:j + word_len]
            if word not in words_set:
                matched_words = []
                break
            else:
                matched_words.append(word)

        if len(matched_words) == len(words):
            result.append(i)

    return result