Leetcode 417：太平洋大西洋流水问题

74 阅读 0 评论 49 点赞

我是靠谱客的博主精明小鸭子，这篇文章主要介绍Leetcode 417：太平洋大西洋流水问题，现在分享给大家，希望可以做个参考。

题目描述

给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界，而“大西洋”处于大陆的右边界和下边界。

规定水流只能按照上、下、左、右四个方向流动，且只能从高到低或者在同等高度上流动。

请找出那些水流既可以流动到“太平洋”，又能流动到“大西洋”的陆地单元的坐标。

提示：

输出坐标的顺序不重要
m 和 n 都小于150

示例：

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
给定下面的 5x5 矩阵:
太平洋 ~
~
~
~
~
~
1
2
2
3
(5) *
~
3
2
3
(4) (4) *
~
2
4
(5)
3
1
*
~ (6) (7)
1
4
5
*
~ (5)
1
1
2
4
*
*
*
*
*
* 大西洋
返回:
[[0, 4], [1, 3], [1, 4], [2, 2], [3, 0], [3, 1], [4, 0]] (上图中带括号的单元).

解题思路

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
class Solution {
public:
int m,n;
void dfs(vector<vector<int>>& matrix,vector<vector<int>>& h,int i,int j,int pre){
if(i>=0&&i<m&&j>=0&&j<n&&h[i][j]==0&&matrix[i][j]>=pre){
pre = matrix[i][j];
h[i][j] = 1;
dfs(matrix,h,i-1,j,pre);
dfs(matrix,h,i,j-1,pre);
dfs(matrix,h,i+1,j,pre);
dfs(matrix,h,i,j+1,pre);
}
}
vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
if(matrix.size() == 0 || matrix[0].size() == 0) return matrix;
m = matrix.size();
n = matrix[0].size();
vector<vector<int>> d(m,vector<int>(n));
vector<vector<int>> t(m,vector<int>(n));
vector<vector<int>> ans;
for(int i=0;i<m;++i){
dfs(matrix,t,i,0,matrix[i][0]);
dfs(matrix,d,i,n-1,matrix[i][n-1]);
}
for(int j=0;j<n;++j){
dfs(matrix,t,0,j,matrix[0][j]);
dfs(matrix,d,m-1,j,matrix[m-1][j]);
}
for(int i=0;i<m;++i){
for(int j=0;j<n;++j){
if(d[i][j]==1 && t[i][j]==1){
vector<int> tmp = {i,j};
ans.push_back(tmp);
}
}
}
return ans;
}
};