每日一题-LeetCode417-太平洋大西洋水流问题-深搜-洪水填充-CSDN博客

本文链接：https://blog.csdn.net/Mr_Ghost812/article/details/124476117

Note：

正着写就超时了，多加个标记数组优化一下还没优化成

改成反着写，从贴着海的地方遍历过去，标记一下每个位置是不是能从这两个海过去

最后在遍历一边Push答案数组

代码如下：

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

    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        vector<vector<int>> ans;
        w = heights;
        n = w.size();
        if(!n)  return ans;
        m = w[0].size();
        st = vector<vector<int>>(n, vector<int>(m));
        
        for(int i = 0; i < n; i ++) dfs(i, 0, 1);
        for(int i = 0; i < m; i ++) dfs(0, i, 1);
        for(int i = 0; i < n; i ++) dfs(i, m - 1, 2);
        for(int i = 0; i < m; i ++) dfs(n - 1, i, 2);
    
        for(int i = 0; i < n; i ++)
            for(int j = 0; j < m; j ++)
                if(st[i][j] == 3)
                    ans.push_back({i, j});
        
        return ans;
    }

    void dfs(int x, int y, int t) {
        // x y 是不是从 t 走过来过
        if (st[x][y] & t) return;
        st[x][y] |= t;
        for (int i = 0; i < 4; i ++ ) {
            int a = x + dx[i], b = y + dy[i];
            if (a >= 0 && a < n && b >= 0 && b < m && w[a][b] >= w[x][y])
                dfs(a, b, t);
        }
    }
};