给定一个 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]] (上图中带括号的单元).
思路:dfs四个方向,找到可以流过的所有的点。两次dfs分别判断能否流到太平洋/大西洋。
class Solution {
// dfs题。
public List<List<Integer>> pacificAtlantic(int[][] matrix) {
int m = matrix.length;
if(m==0){
return new ArrayList();
}
int n = matrix[0].length;
List<List<Integer>> res = new ArrayList();
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
boolean[][] mark1 = new boolean[m][n];
boolean[][] mark2 = new boolean[m][n];
if(dfsTp(matrix,i,j,mark1)&&dfsDx(matrix,i,j,mark2)){
List<Integer> temp = new ArrayList<>();
temp.add(i);
temp.add(j);
res.add(temp);
};
}
}
return res;
}
private boolean dfsTp(int[][] matrix, int i, int j, boolean[][] mark) {
if(i==0||j==0){
return true;
}
mark[i][j] = true;
boolean b1 = false,b2 = false,b3 = false,b4 = false;
if(j<matrix[0].length-1&&!mark[i][j+1]&&matrix[i][j]>=matrix[i][j+1]){
b1 = dfsTp(matrix,i,j+1, mark);
if(b1==true){
return true;
}
}
if(i<matrix.length-1&&!mark[i+1][j]&&matrix[i][j]>=matrix[i+1][j]){
b2= dfsTp(matrix,i+1,j, mark);
if(b2==true){
return true;
}
}
if(!mark[i-1][j]&&matrix[i][j]>=matrix[i-1][j]){
b3= dfsTp(matrix,i-1,j, mark);
if(b3==true){
return true;
}
}
if(!mark[i][j-1]&&matrix[i][j]>=matrix[i][j-1]){
b4= dfsTp(matrix,i,j-1, mark);
if(b4==true){
return true;
}
}
return false;
}
private boolean dfsDx(int[][] matrix, int i, int j, boolean[][] mark) {
if(i==matrix.length-1||j==matrix[0].length-1){
return true;
}
mark[i][j] = true;
boolean b1 = false,b2 = false,b3 = false,b4 = false;
if(!mark[i][j+1]&&matrix[i][j]>=matrix[i][j+1]){
b1 = dfsDx(matrix,i,j+1, mark);
if(b1==true){
return true;
}
}
if(!mark[i+1][j]&&matrix[i][j]>=matrix[i+1][j]){
b2= dfsDx(matrix,i+1,j, mark);
if(b2==true){
return true;
}
}
if(i>0&&!mark[i-1][j]&&matrix[i][j]>=matrix[i-1][j]){
b3= dfsDx(matrix,i-1,j, mark);
if(b3==true){
return true;
}
}
if(j>0&&!mark[i][j-1]&&matrix[i][j]>=matrix[i][j-1]){
b4= dfsDx(matrix,i,j-1, mark);
if(b4==true){
return true;
}
}
return false;
}
}