搜索算法------DFS练习2

1. 题目

2. 思路和题解

从题目中可以看出，如果一个格子上有雨水，那么就可以流到周围比他高度低的单元格，如果单元格和海洋相邻，那么雨水也会流入海洋。总而言之一句话就是水从高处流向低处。从这里的流向可以联想到深度优先搜索这个算法。但是这里我们任意选择单元格去进行搜索的时候，按照正常的思路，后面会不可避免的遍历到重复的单元格，这样会大大的增加搜索时间。所以要换个思路，既然雨水从高流向低，并且靠近海洋的会直接流入海洋，那么我们在采用深度优先搜索的时候，下一个就要去找更大的单元格，利用反向搜索的思路去解决问题。
在进行反向搜索的时候，如果一个单元格既可以从太平洋反向到达，又可以从大西洋反向到达，那么这个单元格就是我们需要的单元格。
代码实现如下:

class Solution {
    static int[][] dirs = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
    int[][] heights;
    int m, n;

    public List<List<Integer>> pacificAtlantic(int[][] heights) {
        this.heights = heights;
        this.m = heights.length;
        this.n = heights[0].length;
        boolean[][] pacific = new boolean[m][n];
        boolean[][] atlantic = new boolean[m][n];
        for (int i = 0; i < m; i++) {
            dfs(i, 0, pacific);
        }
        for (int j = 1; j < n; j++) {
            dfs(0, j, pacific);
        }
        for (int i = 0; i < m; i++) {
            dfs(i, n - 1, atlantic);
        }
        for (int j = 0; j < n - 1; j++) {
            dfs(m - 1, j, atlantic);
        }
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (pacific[i][j] && atlantic[i][j]) {
                    List<Integer> cell = new ArrayList<Integer>();
                    cell.add(i);
                    cell.add(j);
                    result.add(cell);
                }
            }
        }
        return result;
    }

    public void dfs(int row, int col, boolean[][] ocean) {
        if (ocean[row][col]) {
            return;
        }
        ocean[row][col] = true;
        for (int[] dir : dirs) {
            int newRow = row + dir[0], newCol = col + dir[1];
            if (newRow >= 0 && newRow < m && newCol >= 0 && newCol < n && heights[newRow][newCol] >= heights[row][col]) {
                dfs(newRow, newCol, ocean);
            }
        }
    }
}