leetcode 417. Pacific Atlantic Water Flow

原题地址

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

题意

给定一个数字矩阵,每个位置的数字表示这个位置水面的高度,高度大于或等于邻近位置的水面 则 水能从该位置流入邻近位置。找出这个矩阵中既能流向太平洋(矩阵的左边界和上边界)又能流向大西洋(矩阵的右边界和下边界)的位置坐标。

思路

不知道是不是之前都在看动态规划的题。。看到这题第一个思路就是动态规划 [跪了]
想用动态规划就是和那种找出路径数目的方法类似,分别对能通往太平洋的和大西洋的做一次,最后找出交集。但是是有问题的,一个位置能否流向目的地不取决于周围的位置能否流向目的地。。只要足够高就可以了

代码

discuss里的代码

visited表示能到达目的地的情况,初值为0,用一个| 操作来更新自己的值。
从矩阵的左边和上边出发,携带值1进行dfs,从右边和下边出发, 携带值2进行dfs。dfs里的逻辑是,如果当前位置比上一个位置的值更低直接返回(不通),否则当前位置的visited值|上前一个位置的值。
到最后如果两个大洋都能到的位置,其visited值应该为01|10=11=3(10进制),找出所有这样的位置,返回即可。
妙啊×3

class Solution {
public:
    vector<pair<int, int>> res;
    vector<vector<int>> visited;
    void dfs(vector<vector<int>>& matrix, int x, int y, int pre, int preval){
        if (x < 0 || x >= matrix.size() || y < 0 || y >= matrix[0].size()  
                || matrix[x][y] < pre || (visited[x][y] & preval) == preval) 
            return;
        visited[x][y] |= preval;
        if (visited[x][y] == 3) res.push_back({x, y});
        dfs(matrix, x + 1, y, matrix[x][y], visited[x][y]); dfs(matrix, x - 1, y, matrix[x][y], visited[x][y]);
        dfs(matrix, x, y + 1, matrix[x][y], visited[x][y]); dfs(matrix, x, y - 1, matrix[x][y], visited[x][y]);
    }

    vector<pair<int, int>> pacificAtlantic(vector<vector<int>>& matrix) {
        if (matrix.empty()) return res;
        int m = matrix.size(), n = matrix[0].size();
        visited.resize(m, vector<int>(n, 0));
        for (int i = 0; i < m; i++) {
            dfs(matrix, i, 0, INT_MIN, 1);
            dfs(matrix, i, n - 1, INT_MIN, 2);
        }
        for (int i = 0; i < n; i++) {
            dfs(matrix, 0, i, INT_MIN, 1);
            dfs(matrix, m - 1, i, INT_MIN, 2);
        }
        return res;
    }
};

自己的原本的错误代码

class Solution {
  public:
    vector<pair<int, int>> pacificAtlantic(vector<vector<int>>& matrix) {
        vector<pair<int, int>> temp;
        int m  = matrix.size();
        if (m == 0) return temp;
        int n = matrix[0].size();

        vector<vector<bool>> reachP(m, vector<bool>(n, false));
        vector<vector<bool>> reachA(m, vector<bool>(n, false));
        vector<pair<int, int>> result;
        for (int i = 0; i < m; i++) {
            reachP[i][0] = true;
            reachA[i][n - 1] = true;
        }
        for (int j = 0; j < n; j++) {
            reachP[0][j] = true;
            reachA[m - 1][j] = true;
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if ((matrix[i][j] >= matrix[i - 1][j] && reachP[i - 1][j]) || (matrix[i][j] >= matrix[i][j - 1] && reachP[i][j - 1])) {
                    reachP[i][j] = true;
                }
            }
        }
        for (int i = m - 1; i >= 1; i--) {
            for (int j = n - 1; j >= 1; j--) {
                if (i + 1 < m) {
                    reachP[i][j] = reachP[i][j] || (reachP[i + 1][j] && matrix[i][j] >= matrix[i + 1][j]);
                }
                if (j + 1 < m) {
                    reachP[i][j] = reachP[i][j] || (reachP[i][j + 1] && matrix[i][j] >= matrix[i][j + 1]);
                }
            }
        }
        for (int i = m - 2; i >= 0; i--) {
            for (int j = n - 2; j >= 0; j--) {
                if ((matrix[i][j] >= matrix[i + 1][j] && reachA[i + 1][j]) || (matrix[i][j] >= matrix[i][j + 1] && reachA[i][j + 1])) {
                    reachA[i][j] = true;
                }
            }
        }
        for (int i = 0; i < m - 1; i++) {
            for (int j = 0; j < n - 1; j++) {
                if (i - 1 >= 0) {
                    reachA[i][j] = reachA[i][j] || (reachA[i - 1][j] && matrix[i][j] >= matrix[i - 1][j]);
                }
                if (j - 1 >= 0) {
                    reachA[i][j] = reachA[i][j] || (reachA[i][j - 1] && matrix[i][j] >= matrix[i][j - 1]);
                }
            }
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (reachA[i][j] && reachP[i][j]) result.push_back(make_pair(i, j));
            }
        }
        return result;
    }
};
TIM截图20180111011058.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 我是一颗蒲公英的种子,谁也不知道我的快乐和悲伤,在这个世界上,能懂我的人是你,爱我的人依然是你。 外面的世界很精彩...
    芳菲燕阅读 185评论 0 2
  • 文|琉月。 看这本书的心情,和推荐序中许乃涵说的是一样的。连美恩的书适合不刹车地一路冲到终点,读完只觉得意犹未尽。...
    慕卿月阅读 1,116评论 0 1