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> res;
    vector> visited;
    void dfs(vector>& 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> pacificAtlantic(vector>& matrix) {
        if (matrix.empty()) return res;
        int m = matrix.size(), n = matrix[0].size();
        visited.resize(m, vector(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> pacificAtlantic(vector>& matrix) {
        vector> temp;
        int m  = matrix.size();
        if (m == 0) return temp;
        int n = matrix[0].size();

        vector> reachP(m, vector(n, false));
        vector> reachA(m, vector(n, false));
        vector> 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;
    }
};
leetcode 417. Pacific Atlantic Water Flow_第1张图片
TIM截图20180111011058.png

你可能感兴趣的:(leetcode 417. Pacific Atlantic Water Flow)