题目
有一个 m × n 的矩形岛屿,与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界,而 “大西洋” 处于大陆的右边界和下边界。
这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights , heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。
岛上雨水较多,如果相邻单元格的高度 小于或等于 当前单元格的高度,雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
返回网格坐标 result 的 2D 列表 ,其中 result[i] = [ri, ci] 表示雨水从单元格 (ri, ci) 流动 既可流向太平洋也可流向大西洋 。
示例
输入: heights = [[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]]
输入: heights = [[2,1],[1,2]] 输出: [[0,0],[0,1],[1,0],[1,1]]
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/pacific-atlantic-water-flow 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:DFS
?题解:https://leetcode-cn.com/problems/pacific-atlantic-water-flow/solution/shui-wang-gao-chu-liu-by-xiaohu9527-xxsx/
Java实现
class Solution {
int m, n;
int[] directions = new int[]{-1, 0, 1, 0, -1};
int[][] heights;
public List<List<Integer>> pacificAtlantic(int[][] heights) {
this.heights = heights;
m = heights.length;
n = heights[0].length;
boolean[][] visitedPo = new boolean[m][n];
boolean[][] visitedAo = new boolean[m][n];
for (int i = 0; i < m; i++) {
track_back(i, 0, visitedPo);
track_back(i, n - 1, visitedAo);
}
for (int j = 0; j < n; j++) {
track_back(0, j, visitedPo);
track_back(m - 1, j, visitedAo);
}
List<List<Integer>> res = new ArrayList<>();
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (visitedPo[i][j] && visitedAo[i][j]) {
res.add(Arrays.asList(i, j));
}
}
}
return res;
}
public void track_back(int i, int j, boolean[][] visited) {
if (visited[i][j]) return;
visited[i][j] = true;
for (int k = 0; k < directions.length - 1; k++) {
int x = i + directions[k];
int y = j + directions[k + 1];
if (x < m && x >= 0 && y < n && y >= 0 && heights[x][y] >= heights[i][j]) {
track_back(x, y, visited);
}
}
}
}
|