Learn DSA
Depth-First Search
Greedy Algorithms
Depth-First Search
Pacific Atlantic Water Flow
medium
DESCRIPTION (credit Leetcode.com)
You are given an m x n matrix of non-negative integers representing a grid of land. Each value in the grid represents the height of that piece of land.
The Pacific Ocean touches the left and top edges of the matrix, while the Atlantic Ocean touches the right and bottom edges. Water can only flow from a cell to its neighboring cells directly north, south, east, or west, but only if the height of the neighboring cell is equal to or lower than the current cell.
Write a function to return a list of grid coordinates (i, j) where water can flow to both the Pacific and Atlantic Oceans. Water can flow from all cells directly adjacent to the ocean into that ocean.
Example 1:
Input:
Output:
Explanation
Approach 1: Brute Force Approach
class Solution:def pacificAtlantic(self, matrix):if not matrix or not matrix[0]:return []rows, cols = len(matrix), len(matrix[0])pacific = set()atlantic = set()# Try to find a path from r, c to the Pacific or Atlantic ocean# via neighboring cells with lower heightsdef dfs(start_r, start_c, r, c, visited):if (r, c) in visited:returnvisited[(r, c)] = Trueif r == 0 or c == 0:pacific.add((start_r, start_c))if r == rows - 1 or c == cols - 1:atlantic.add((start_r, start_c))directions = [(1, 0), (-1, 0), (0, 1), (0, -1)]for dr, dc in directions:nr, nc = r + dr, c + dcif 0 <= nr < rows and 0 <= nc < cols and matrix[nr][nc] <= matrix[r][c]:dfs(start_r, start_c, nr, nc, visited)visited[(nr, nc)] = False# Perform full DFS from each cell.for r in range(rows):for c in range(cols):visited = {}dfs(r, c, r, c, visited)return list(pacific & atlantic)
Approach 2: Boundary DFS
Solution
class Solution:def pacific_atlantic_flow(self, grid):if not grid or not grid[0]:return []rows, cols = len(grid), len(grid[0])# Step 1: Initialize empty setspacific_reachable = set()atlantic_reachable = set()def dfs(r, c, reachable):reachable.add((r, c))for dr, dc in [(1,0), (-1,0), (0,1), (0,-1)]:nr, nc = r + dr, c + dcif 0 <= nr < rows and 0 <= nc < cols:if (nr, nc) not in reachable and grid[nr][nc] >= grid[r][c]:dfs(nr, nc, reachable)# initializes DFS from all cells in the Atlantic and Pacific Oceans# Note how we share a single visited set# across DFS calls that originate from the same oceanfor r in range(rows):dfs(r, 0, pacific_reachable)dfs(r, cols - 1, atlantic_reachable)for c in range(cols):dfs(0, c, pacific_reachable)dfs(rows - 1, c, atlantic_reachable)# return the intersection of both sets.return list(pacific_reachable & atlantic_reachable)
Complexity Analysis
Login to track your progress
Your account is free and you can post anonymously if you choose.