solve(w09): 76. Minimum Window Substring

Author: seungriyou

minimum-window-substring/seungriyou.py

@@ -0,0 +1,120 @@
+# https://leetcode.com/problems/minimum-window-substring/
+
+class Solution:
+    def minWindow1(self, s: str, t: str) -> str:
+        """
+        [Complexity]
+            - TC: O(m * k + n) (k = is_included() = len(set(t)))
+            - SC: O(k)
+
+        [Approach]
+            다음과 같은 counter를 유지하며, two pointer로 window를 이동하며 min window substring을 트래킹한다.
+                - 생성 시) t의 모든 문자에 대해 ++
+                - pointer 이동 시) window에 포함되는 문자 중 t에 속하는 문자에 대해 --
+        """
+        m, n = len(s), len(t)
+
+        # early stop
+        if m < n:
+            return ""
+
+        # t에 대한 counter 생성
+        counter = {}
+        for _t in t:
+            if _t not in counter:
+                counter[_t] = 0
+            counter[_t] += 1
+
+        # t의 모든 문자가 window에 포함되는지 확인하는 함수
+        def is_included():
+            # counter의 모든 값이 0 이하이면, t의 모든 문자가 window에 포함되는 것 (중복 포함)
+            return all(c <= 0 for c in counter.values())
+
+        lo, min_len = 0, 1e6
+        res = ""
+
+        for hi in range(m):
+            # 현재 window에 포함된 문자를 counter에 반영
+            if s[hi] in counter:
+                counter[s[hi]] -= 1
+
+            # t의 모든 문자가 window에 포함되어있다면(= counter의 모든 값이 <= 0이면),
+            # counter 값이 음수 ~ 0이 될 때까지 lo 증가시키면서 (min window를 구해야하므로) counter 업데이트
+            if is_included():
+                while True:
+                    # window의 첫 번째 문자가 t에 속하는 문자라면
+                    if s[lo] in counter:
+                        # counter에서의 값이 0이면, 더이상 lo를 오른쪽으로 줄일 수 없음
+                        if counter[s[lo]] == 0:
+                            break
+                        # coutner에서의 값이 음수라면, lo를 오른쪽으로 줄일 수 있으므로 counter 반영
+                        else:
+                            counter[s[lo]] += 1
+                    # lo를 오른쪽으로 한 칸 이동
+                    lo += 1
+
+            # (1) t의 모든 문자가 window에 포함되어 있고 (2) 현재 window의 length가 min_len 보다 작다면
+            # window substring 업데이트
+            if is_included() and (new_len := hi - lo + 1) < min_len:
+                min_len = new_len
+                res = s[lo:hi + 1]
+
+        return res
+
+    def minWindow(self, s: str, t: str) -> str:
+        """
+        [Complexity]
+            - TC: O(m + n)
+            - SC: O(k)
+
+        [Approach]
+            위의 풀이에서 is_included()를 실행하는 데에 O(k)이 소요되므로,
+            (t에 속하는 문자 중, 현재 window에 모두 포함되는 문자의 종류 개수)를 트래킹하는 변수를 이용하여 최적화 한다.
+            즉, have == len(counter)라면 is_included()인 것과 동일하다.
+        """
+        m, n = len(s), len(t)
+
+        # early stop
+        if m < n:
+            return ""
+
+        # t에 대한 counter 생성
+        counter = {}
+        for _t in t:
+            if _t not in counter:
+                counter[_t] = 0
+            counter[_t] += 1
+
+        # counter의 값을 모두 확인하는 is_include() 최적화
+        # have = t에 속하는 문자 중, 현재 window에 모두 포함되는 문자의 종류 개수
+        # have == need라면, is_included() 인 것과 같음
+        have, need = 0, len(counter)
+        lo, min_len = 0, 1e6
+        res = ""
+
+        for hi in range(m):
+            # 현재 window에 포함된 문자를 counter에 반영
+            if s[hi] in counter:
+                counter[s[hi]] -= 1
+                # 현재 window에 해당 문자가 t에 존재하는 개수만큼 들어와있다면, have++
+                if counter[s[hi]] == 0:
+                    have += 1
+
+            # t의 모든 문자가 window에 포함되어있는 동안 lo 이동
+            while have == need:
+                # 최소 길이 window substring 갱신
+                if (new_len := hi - lo + 1) < min_len:
+                    min_len = new_len
+                    res = s[lo:hi + 1]
+
+                # lo 이동 전, counter 업데이트
+                if s[lo] in counter:
+                    counter[s[lo]] += 1
+                    # counter의 값이 0 초과가 된다면, 더이상 window에 해당 문자가 모두 들어있지 않다는 것이므로 have--
+                    if counter[s[lo]] > 0:
+                        have -= 1
+
+                # lo를 오른쪽으로 한 칸 이동
+                lo += 1
+
+        return res