力扣417题太平洋大西洋水流问题
题目描述:有一个 m × n 的矩形岛屿,与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界,而 “大西洋” 处于大陆的右边界和下边界。
这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights , heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。
岛上雨水较多,如果相邻单元格的高度 小于或等于 当前单元格的高度,雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
返回 网格坐标 result 的 2D列表 ,其中 result[i] = [ri, ci] 表示雨水可以从单元格 (ri, ci) 流向 太平洋和大西洋 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pacific-atlantic-water-flow
先吐槽下力扣的翻译,真的有时候读了半天题目感觉莫名其妙,有时候会让人血压飙升。这道题目说白了就是给一个二维数组描述一个岛屿各个位置的高度,求雨水既能流入太平洋又能流入大西洋的位置。
思路分析:需要判断每个位置是否能够流入太平洋以及流入大西洋。可以从左边以及上边开始搜索,寻找哪些位置可以流向大西洋;接着从右边以及下边开始搜索,寻找哪些位置可以流向太平洋。我也使用深度优先算法进行代码编写,几乎和官方题解一样,不同的是我会对代码添加一些注释,方便读懂算法。
代码如下
class Solution {
int[][] dirs = new int[][]{{0,-1},{0,1},{1,0},{-1,0}};//用于计算前后左右位置
int row;//岛屿的行数
int column;//岛屿的列数
int[][] arr;//就是heights,只是方便函数调用
public List<List<Integer>> pacificAtlantic(int[][] heights) {
arr = heights;
row = heights.length;
column = heights[0].length;
//创建是否可以流向两洋的布尔数组
boolean[][] pacific = new boolean[row][column];
boolean[][] atlantic = new boolean[row][column];
//从左边界进行搜索
for (int i = 0; i < row; i++) {
dfs(i,0,pacific);
}
//从上边界进行搜索
for (int j = 0; j < column; j++) {
dfs(0,j,pacific);
}
//从右边界进行搜索
for (int i = 0; i <row; i++) {
dfs(i,column-1,atlantic);
}
//从下边界进行搜索
for (int j = 0; j < column; j++) {
dfs(row-1,j,atlantic);
}
//求数组pacific以及数组atlantic的交集
//交集中的元素就是既能流向大西洋又能流向太平洋
List<List<Integer>> res = new ArrayList<>();
for (int i = 0; i < row; i++) {
for (int j = 0; j < column; j++) {
if(pacific[i][j]&& atlantic[i][j]){
List<Integer> list = new ArrayList<>();
list.add(i);
list.add(j);
res.add(list);
}
}
}
return res;
}
//深度优先算法
private void dfs(int i, int j, boolean[][] ocean) {
//停止条件:如果待搜索的元素是true,那么说明已经搜索了该元素,可以停止
if(ocean[i][j]){
return;
}
//处理该元素,因为能够到达该元素,那么就该元素满足条件
ocean[i][j] = true;
//搜索相邻元素,上下左右
for(int[] dir:dirs){
int newRow = i+dir[0];
int newColumn = j+dir[1];
//相邻元素不能越界,前面的条件,最后一个条件就是找更高的位置,对应ocean[i][j] = true;语句进行理解
if(newRow>=0 && newRow<row && newColumn>=0 && newColumn<column && arr[newRow][newColumn]>=arr[i][j]){
dfs(newRow,newColumn,ocean);
}
}
}
}