Solved Pacific Atlantic Water Flow problem with code review

Author: KwonNayeon

pacific-atlantic-water-flow/KwonNayeon.py

@@ -6,18 +6,19 @@
 - 0 <= heights[r][c] <= 10^5
 
 Time Complexity: O(m*n) 
-- 각 셀을 최대 한 번씩만 방문함
+- 각 셀을 최대 한 번만 방문함
 
 Space Complexity: O(m*n)
-- visited sets(pacific, atlantic)가 최대 m*n 크기
-- 재귀 호출 스택도 최대 m*n 깊이 가능
+- visited sets(pacific, atlantic)은 최대 m*n 크기
+- 재귀 호출 스택도 최대 m*n까지 가능
 
 풀이방법:
-1. Pacific(좌측상단)과 Atlantic(우측하단)의 경계에서 시작함
-2. DFS로 현재 높이보다 높거나 같은 인접 셀로만 이동함 (물은 위 -> 아래로 흐르지만, 거꾸로 접근했으니까)
+1. Pacific(좌측상단)과 Atlantic(우측하단)의 경계에서 시작
+2. DFS로 현재 높이보다 높거나 같은 인접 셀로만 이동 (역방향 접근)
 3. 각 바다에서 도달 가능한 셀들을 Set에 저장
-4. 두 Set의 교집합이 정답 (양쪽 바다로 모두 흐를 수 있는 지점들)
+4. 두 Set의 교집합으로 양쪽 바다 모두 도달 가능한 지점 반환
 """
+# 원본 코드
 class Solution:
     def pacificAtlantic(self, heights: List[List[int]]) -> List[List[int]]:
         if not heights:
@@ -51,3 +52,43 @@ def dfs(r, c, visited):
             dfs(r, cols-1, atlantic)
 
         return list(pacific & atlantic)
+
+# 개선된 코드
+class Solution:
+    def pacificAtlantic(self, heights: List[List[int]]) -> List[List[int]]:
+        rows, cols = len(heights), len(heights[0])
+
+        # 각 바다에서 도달 가능한 위치 저장
+        pacific = set()
+        atlantic = set()
+
+        def dfs(r, c, visited):
+            # 현재 위치 방문 처리
+            visited.add((r, c))
+
+            # 네 방향 탐색 (오른쪽, 위, 왼쪽, 아래)
+            for dr, dc in [(0, 1), (-1, 0), (0, -1), (1, 0)]:
+                new_r, new_c = r + dr, c + dc
+
+                # 경계 체크, 방문x 체크, 높이 체크
+                if (0 <= new_r < rows and
+                    0 <= new_c < cols and
+                    (new_r, new_c) not in visited and
+                    heights[new_r][new_c] >= heights[r][c]):
+
+                    dfs(new_r, new_c, visited)
+
+        # Pacific 경계에서 시작 (위 + 왼쪽)
+        for c in range(cols):
+            dfs(0, c, pacific)
+        for r in range(rows):
+            dfs(r, 0, pacific)
+
+        # Atlantic 경계에서 시작 (아래 + 오른쪽)
+        for c in range(cols):
+            dfs(rows-1, c, atlantic)
+        for r in range(rows):
+            dfs(r, cols-1, atlantic)
+
+        # 양쪽 바다 모두 도달 가능한 위치 반환
+        return [[r, c] for r, c in pacific & atlantic]