【Leetcode】417. Pacific Atlantic Water Flow(代码)

219 阅读1分钟

题目地址: leetcode.com/problems/pa…

给定一个矩阵,矩阵左边界和上边界代表太平洋,右边界和下边界代表大西洋,从矩阵某一点开始,可以沿着递降路径一路走下去,如果存在某两条路径分别与太平洋和大西洋接通,那么这个点就是我们要找的。返回所有要找的点的坐标。主要的思想是floodfill,可以先从太平洋和大西洋边上的每个点出发,沿着一条递升路径把沿途的点都标记为可以连通到太平洋(或者大西洋),然后返回同时能到两个洋的点的坐标即可。代码如下:

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
    public List<List<Integer>> pacificAtlantic(int[][] matrix) {
        List<List<Integer>> res = new ArrayList<>();
        // 判空
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
            return res;
        }
        
        int m = matrix.length, n = matrix[0].length;
        // 标记某个位置是否被访问过
        boolean[][] visited = new boolean[m][n];
        // 标记某个点是否能连通到太平洋和大西洋,如果仅能连通到太平洋则赋值为1,如果仅能连通到大西洋则赋值为2,如果都能,则赋值为3
        int[][] canflow = new int[m][n];
        // 标记四个方向
        int[][] d = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
    	// 开始搜索能连通到太平洋的点
        for (int i = 0; i < m; i++) {
            dfs(matrix, i, 0, visited, canflow, d, 1);
        }
        for (int j = 0; j < n; j++) {
            dfs(matrix, 0, j, visited, canflow, d, 1);
        }
    	
    	// 此时能连通到太平洋的点已经被标记了,这时候要重置visited数组,然后继续搜索能连通到大西洋的点
        for (int i = 0; i < visited.length; i++) {
            Arrays.fill(visited[i], false);
        }
        
        for (int i = 0; i < m; i++) {
            dfs(matrix, i, n - 1, visited, canflow, d, 2);
        }
        for (int j = 0; j < n; j++) {
            dfs(matrix, m - 1, j, visited, canflow, d, 2);
        }
    
    	// 将能同时连到两个洋的点的下标加入res返回
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (canflow[i][j] == 3) {
                    res.add(Arrays.asList(i ,j));
                }
            }
        }
        
        return res;
    }
    
    private void dfs(int[][] matrix, int x, int y, boolean[][] visited,
                     int[][] canflow, int[][] d, int ocean) {
        // 先将当前点标记为访问过
        visited[x][y] = true;
        // 如果当前点能连通太平洋,此时正在搜索能连通大西洋的点,说明该点两个洋都能连通,所以赋值为3;
        // 否则如果当前点未被标记过,则赋值为能连通的大洋编号
        if (canflow[x][y] == 1 && ocean == 2) {
            canflow[x][y] = 3;
        } else if (canflow[x][y] == 0) {  // 这个判断是必要的,否则赋值为3之后再搜索的时候,新的赋值会覆盖从而产生错误
            canflow[x][y] = ocean;
        }
        
        for (int i = 0; i < 4; i++) {
            int newX = x + d[i][0];
            int newY = y + d[i][1];
            // 如果将要走的点在范围内,并且未走过,而且路径是上升的,那么就可以枚举
            if (newX >= 0 && newX < matrix.length && newY >= 0 && newY < matrix[0].length
                && !visited[newX][newY] && matrix[newX][newY] >= matrix[x][y]) {
                dfs(matrix, newX, newY, visited, canflow, d, ocean);
            }
        }
    }
}

相关链接:

  1. (7条消息) LC DFS、BFS与图论_edWard的博客-CSDN博客

  2. happygirlzt

  3. github.com/h03147/Algo…