1. 程式人生 > 其它 >力扣 leetcode 424. 替換後的最長重複字元 (python)

力扣 leetcode 424. 替換後的最長重複字元 (python)

技術標籤:pythonleetcode字串python演算法leetcode雙指標

Topic

給你一個僅由大寫英文字母組成的字串,你可以將任意位置上的字元替換成另外的字元,總共可最多替換 k 次。在執行上述操作後,找到包含重複字母的最長子串的長度。
注意:字串長度 和 k 不會超過 104。

Example_1

輸入:s = “ABAB”, k = 2
輸出:4
解釋:用兩個’A’替換為兩個’B’,反之亦然。

Example_2

輸入:s = “AABABBA”, k = 1
輸出:4
解釋:
將中間的一個’A’替換為’B’,字串變為 “AABBBBA”。
子串 “BBBB” 有最長重複字母, 答案為 4。

Solution

可以利用滑動視窗的方法(雙指標)解決本題
首先設定一個字典
用於記錄視窗中的字母個數
設定一個length記錄為滑動視窗最大長
初始化左側指標left為0

不斷遍歷s
right為遍歷時的索引,val為s中的字母
將對應的字母數量+1

當視窗中的個數減字典中最大的數(能交換的個數)比k大
那麼左側指標(滑窗左邊緣)就需要右移
並維護的字典中滑動視窗左側指標對應的字母減一(字母數量減少)
同時最大長度length也與選擇當前視窗數量和之前最大長度中的最大值

最後返回length長即是最長重複字母數量

Code

class Solution:
    def characterReplacement
(self, s: str, k: int) -> int: dic = {} length = 0 left = 0 for right, ch in enumerate(s): dic[ch] = dic.get(ch, 0) + 1 if (right - left + 1) - max(dic.values()) > k: dic[s[left]] -= 1 left += 1 length =
max(length, right - left + 1) return length

Result

在這裡插入圖片描述