1. 问题描述:
给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界,而“大西洋”处于大陆的右边界和下边界。规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。
提示:
输出坐标的顺序不重要
m 和 n 都小于150
示例:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pacific-atlantic-water-flow
2. 思路分析:
分析题目可以知道这是一道经典的搜索题目,可以使用dfs或者是bfs搜索,这里使用dfs搜索解决,一个比较简单的想法是使用两遍dfs搜索,第一遍dfs搜索能够到达太平洋的区域,第二遍dfs搜索能够到达大西洋的区域,当同时被两遍dfs搜索到的位置那么就是既可以到达太平洋又可以到达大西洋的的位置,但是我们在dfs搜索的时候怎么样标记这些位置呢?这里可以使用二进制来标记当前搜索到达的是太平洋还是大西洋,1表示到达的是太平洋,2表示到达的是大西洋。为了方便搜索我们可以从边界开始搜索,也即第一行,第一列,最后一行,最后一列分别表示太平洋和大西洋的能够到达的位置,从边界开始搜索的时候那么下一个位置的高度就需要大于等于当前位置的高度。怎么样判断当前的位置是否被搜索过呢,我们需要声明一个与矩阵同样大小的状态列表st来标记当前位置的状态,并且在递归方法中传递一个int类型的变量t来表示当前dfs搜索的是太平洋还是大西洋,当st[x][y] & t != 0说明当前位置已经被搜索过,并且如果第一次搜索的是太平洋第二次搜索的是大西洋那么st[x][y] & t = 1 & 2 = 0所以还是可以搜索到的,最终所有能够到达太平洋和大西洋的位置都会被标记为3。
3. 代码如下:
from typing import List
class Solution:
# 表示右左下上四个方向
pos = [[0, 1], [0, -1], [1, 0], [-1, 0]]
def dfs(self, x: int, y: int, w: List[List[int]], st: List[List[int]], t: int):
# 表示之前已经搜索过这个状态
if st[x][y] & t: return
# 标记当前搜索的是哪一种,是太平洋还是大西洋
st[x][y] |= t
for i in range(4):
x1, y1 = x + self.pos[i][0], y + self.pos[i][1]
# 反向考虑需要下一个位置需要大于等于自己的高度
if 0 <= x1 < len(w) and 0 <= y1 < len(w[0]) and w[x1][y1] >= w[x][y]:
self.dfs(x1, y1, w, st, t)
# 使用二进制来表示被搜索过的状态, 1表示搜索太平洋, 2表示搜索大西洋
def pacificAtlantic(self, heights: List[List[int]]) -> List[List[int]]:
r, c = len(heights), len(heights[0])
if r == 0 or c == 0: return []
# st表示搜索过的状态
st = [[0] * c for i in range(r)]
for i in range(r): self.dfs(i, 0, heights, st, 1)
for i in range(c): self.dfs(0, i, heights, st, 1)
for i in range(r): self.dfs(i, c - 1, heights, st, 2)
for i in range(c): self.dfs(r - 1, i, heights, st, 2)
# 遍历状态
res = list()
for i in range(r):
for j in range(c):
# 两次dfs都搜索到了
if st[i][j] == 3:
res.append([i, j])
return res