力扣 leetcode 424. 替換後的最長重複字元 (python)
阿新 • • 發佈:2021-02-03
技術標籤: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