2022/04/27 leetcode 每日一题 417. 太平洋大西洋水流问题

1.题目

2.求解

这个题目的描述真的很让人迷惑，建议大家就是看看英文的描述：

        其实题目想说的就很简单，从每个节点判断能否到达两边边界。如果将n*m的节点全遍历完，那所要花费的时间太长了。因此可以从边界开始遍历，分别从太平洋和 和大西洋，分两部分遍历。即从左边和上面能到达的点，和从右边和下面能到达的点，每次只往等高或更高的点前进，能到则cnt++。两次遍历分别维护一个visit数组，进行记忆化搜索，加快运行速度，但是会消耗一部分内存。

3.代码

class Solution {
bool visit[205][205];
int cnt[205][205];
int num[4][2] = {1, 0, 0, 1, -1, 0, 0, -1};
vector<vector<int>> ans;
vector<vector<int>> h;
int n, m;
public:
    void dfs(int x, int y){
        if(visit[x][y] == true) return;
        visit[x][y] = true;
        cnt[x][y]++;
        if(cnt[x][y] == 2){
            // cout<<x<<" "<<y<<endl;
            ans.push_back(vector<int>{x, y});
        }
        for(int i = 0; i < 4; i++){
            int nextx = x + num[i][0];
            int nexty = y + num[i][1];
            if(nextx >= n || nextx < 0) continue;
            if(nexty >= m || nexty < 0) continue;
            // cout<<nextx<<nexty<<endl;
            if(h[nextx][nexty] >= h[x][y]) {
                dfs(nextx, nexty);
            }
        }
    }
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        int i;
        h = heights;
        n = heights.size();
        m = heights[0].size();
        memset(cnt, 0, sizeof(cnt));
        memset(visit, false, sizeof(visit));
        ans.clear();
        for(i = 0; i < m; i++)dfs(0, i);
        for(i = 0; i < n; i++)dfs(i, 0);
        memset(visit, false, sizeof(visit));
        for(i = 0; i < m; i++)dfs(n - 1, i);
        for(i = 0; i < n; i++)dfs(i, m - 1);
        return ans;
    }
};