【Leetcode】417. Pacific Atlantic Water Flow

题目地址：

https://leetcode.com/problems/pacific-atlantic-water-flow/

给定一个矩阵，矩阵左边界和上边界代表太平洋，右边界和下边界代表大西洋，从矩阵某一点开始，可以沿着递降路径一路走下去，如果存在某两条路径分别与太平洋和大西洋接通，那么这个点就是我们要找的。返回所有要找的点的坐标。主要的思想是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);
            }
        }
    }
}

时空复杂度$O(mn)$。