417. 太平洋大西洋水流问题
给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界,而“大西洋”处于大陆的右边界和下边界。
规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。
请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。
提示:
输出坐标的顺序不重要
m 和 n 都小于150
示例:
给定下面的 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]] (上图中带括号的单元).
- 大西洋
-
-
-
-
代码
使用深度优先遍历
class Solution {
public:
//注意一下 在这块传参时vector<vector<int>> &matrix,这块使用引用,不然力扣会判超时。题不难但是我看了好久,就是因为这个超时。
void dfs(vector<vector<int>> &matrix,int row,int col ,int height, vector<vector<int>>& tag) {
int R = matrix.size(), C = matrix[0].size();
if ( row<0||col < 0 || row >= R || col >= C || tag[row][col] == 1 || matrix[row][col] < height)return;
tag[row][col] = 1;
dfs(matrix, row - 1, col, matrix[row][col], tag);
dfs(matrix, row, col - 1, matrix[row][col], tag);
dfs(matrix, row + 1, col, matrix[row][col], tag);
dfs(matrix, row, col + 1, matrix[row][col], tag);
}
vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
if (matrix.empty())return {};
vector<vector<int>>res;
int R = matrix.size(), C = matrix[0].size();
vector<vector<int>>pacific_tag(R, vector<int>(C, 0));//太平洋标签矩阵
vector<vector<int>>atlantic_tag(R, vector<int>(C, 0));//大西洋标签矩阵
//左右边界
for (int i = 0; i < R; ++i) {
dfs(matrix, i, 0, matrix[i][0], pacific_tag);
dfs(matrix, i, C-1, matrix[i][C - 1], atlantic_tag);
}
//上下边界
for (int j = 0; j < C; ++j) {
dfs(matrix, 0, j, matrix[0][j], pacific_tag);
dfs(matrix, R-1,j, matrix[R - 1][j], atlantic_tag);
}
for (int i = 0; i < R; ++i) {
for (int j = 0; j < C; ++j) {
if (pacific_tag[i][j] == 1 && atlantic_tag[i][j] == 1)
res.push_back({ i,j });
}
}
//myprint2(res);
return res;
}
};
测试用例:
int main() {
vector<vector<int>>vec = { {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} };
Solution S;
//vector<vector<int>>vec1(0, vector<int>(0));
S.pacificAtlantic(vec);
return 0;
}
结果:[[0, 4], [1, 3], [1, 4], [2, 2], [3, 0], [3, 1], [4, 0]]