Leetcode 417. 太平洋大西洋水流问题

有一个 m × n 的矩形岛屿，与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界，而 “大西洋” 处于大陆的右边界和下边界。

这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights ， heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。

岛上雨水较多，如果相邻单元格的高度 小于或等于 当前单元格的高度，雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。

返回网格坐标 result 的 2D 列表 ，其中 result[i] = [ri, ci] 表示雨水从单元格 (ri, ci) 流动 既可流向太平洋也可流向大西洋 。

示例 1：

输入: heights = [[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]]
示例 2：

输入: heights = [[2,1],[1,2]]
输出: [[0,0],[0,1],[1,0],[1,1]]

提示：

m == heights.length
n == heights[r].length
1 <= m, n <= 200
0 <= heights[r][c] <= 105

思路：这道题问的就是有多少个格子可以同时流到大西洋和太平洋。也就是逐渐降低至左上两条边或者右下两条边。那我们可以反着思考，就是分别从两条边出发，看能上升至那些边。然后标记一下。

class Solution {
public:
    vector<vector<int>> g;
    int n, m;
    vector<vector<int>> p, a;
    int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};


    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        g = heights;
        n = g.size(), m = g[0].size(); 
        p = a = vector<vector<int>>(n, vector<int>(m, 0));
        
        
        for(int i = 0; i < n; i ++ ) dfs(i, 0, 0); // 前两位是坐标，最后一位是标记正在做哪个大洋的
        for(int i = 0; i < m; i ++ ) dfs(0, i, 0);
        for(int i = 0; i < n; i ++ ) dfs(i, m - 1, 1);
        for(int i = 0; i < m; i ++ ) dfs(n - 1, i, 1);
        vector<vector<int>> res;
        for(int i = 0; i < n; i ++ )
            for(int j = 0; j < m; j ++ )
                if(p[i][j] && a[i][j]) res.push_back({i, j});
        return res;
    }

    void dfs(int x, int y, int t) {
        
        if(x < 0 || x >= n || y < 0 || y >= m) return;
        if((!t && p[x][y]) || (t && a[x][y])) return;
        if(!t) p[x][y] = 1;
        else a[x][y] = 1;
        for(int i = 0; i < 4; i ++ ) {
            int k = x + dx[i], v = y + dy[i];
            if(k >= 0 && k < n && v >= 0 && v < m)
                if(g[x][y] <= g[k][v])
                    dfs(k, v, t);
        }
    }
};