一、滑動窗口概述
滑動窗口,顧名思義就是一個可以滑動的窗口,它可以用來解決一些數組/字符串的子元素問題。
滑動窗口的核心思想是,定義一些指針和索引來表示一些位置,移動窗口,更新狀態,並且在不遺漏任何位置的前提下,找到我們需要的答案。
那麼,程序員在使用滑動窗口算法時,需要如何思考呢?
首先,一定是定義一個窗口,這個窗口通常是一個連續的子區間,然後用變量 left 和 right 表示窗口的左右邊界,再根據問題的需要定義一些其它變量。然後,窗口開始向右滑動,當窗口右端點右移時,也就是 right += 1 時,窗口中多了一個右邊的元素,窗口的和會發生什麼變化?需要減少還是增加?這部分問題需要我們自己去思考。
二、常用滑動窗口算法題目及解法
1. 最小覆蓋子串
題目描述:給定一個字符串 S 和一個字符串 T,請在 S 中找出包含 T 所有字母的最小子串。
解法:使用雙指針和哈希表。首先使用哈希表記錄下字符串 T 中每個字符出現的次數,然後使用雙指針 left 和 right 來表示窗口的左右邊界。當窗口中包含所有字符時,取最小子串,然後左指針右移,不斷更新最小字符串。當窗口中不包含所有字符時,右指針右移。
class Solution:
def minWindow(self, s: str, t: str) -> str:
if not s or not t:
return ""
left, right = 0, 0
cnt = len(t)
need = {}
for i in t:
need[i] = need.get(i, 0) + 1
res = ""
min_len = float("inf")
while right 0:
cnt -= 1
need[s[right]] -= 1
right += 1
while cnt == 0:
if (right - left) < min_len:
min_len = right - left
res = s[left:right]
if s[left] in need:
if need[s[left]] == 0:
cnt += 1
need[s[left]] += 1
left += 1
return res
2. 求和等於K的最長子數組
題目描述:給定一個整數數組 nums 和一個目標值 k,請找到數組中和等於 k 的最長子數組長度。如果不存在任意一個符合條件的子數組,則返回 0。
解法:使用前綴和加哈希表。首先,定義一個哈希表,存儲每個前綴和出現的最早下標。然後,定義一個變量 sum 為前綴和,當 sum – k 在哈希表中存在時,說明從 i 到 j 存在一個子數組和等於 k,將其長度與之前計算的最大長度進行比較,更新答案。
class Solution:
def maxSubArrayLen(self, nums: List[int], k: int) -> int:
if not nums:
return 0
res = 0
prefix_sum = 0
has_dict = {0:-1}
for i in range(len(nums)):
prefix_sum += nums[i]
if prefix_sum - k in has_dict:
res = max(res, i - has_dict[prefix_sum - k])
if prefix_sum not in has_dict:
has_dict[prefix_sum] = i
return res
三、滑動窗口要點總結
1、滑動窗口算法(雙指針算法)是一種固定長度的窗口在數組上滑動,通過雙指針(數組坐標)對數組一個個進行滑動統計,並得出問題的解。
2、滑動窗口的時間複雜度通常為 O(n),比一般的暴力解法要高效許多。
3、當想要在一個數組或是一個字符串中找到一個子區間,滑動窗口算法應該是一個不錯的選擇。
4、在使用滑動窗口算法的過程中,一定要注意邊界問題,並且稍加思考時,可以使用哈希表等數據結構優化算法。
5、掌握了滑動窗口算法的思想和模板,我們可以針對具體問題進行適當的修改和調整,解決更多的問題。
原創文章,作者:小藍,如若轉載,請註明出處:https://www.506064.com/zh-hk/n/151979.html
微信掃一掃
支付寶掃一掃