417. Pacific Atlantic Water Flow

一、题目简述

给定一个m*n的非负矩阵代表大陆中每个单元格的高度，太平洋与矩阵的左、上边相邻，大西洋与矩阵的右、下边相邻。
水流只能沿着四个方向进行流动（上、下、左、右），从一个较高的单元格流向较低或等高的单元格。
寻找既能流到大西洋又能流到太平洋的单元格的坐标。
注意：

1. 返回单元格坐标的顺序不影响答案的正确性。
2. m与n均小于150。

例如：
Given the following 5x5 matrix:

  Pacific ~   ~   ~   ~   ~ 
       ~  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  *
          *   *   *   *   * Atlantic

Return:
[[0, 4], [1, 3], [1, 4], [2, 2], [3, 0], [3, 1], [4, 0]] (positions with parentheses in above matrix).

二、编程思路

本题目可以分为两个部分就行解决，分别找到能够流到大西洋、太平洋单元格的坐标。自然的思路是使用广度优先搜索，例如求能够流到太平洋的单元格坐标时，可以从矩阵的左、上边开始搜索，寻找需要其流经方向上高度高于或等于当前单元格的格子，并将其状态置为可到达。之后从可到达的单元格上做相同搜索即可。
在获得可到达太平洋与大西洋单元格的坐标之后，取其交集就是能够同时达到大西洋和太平洋的单元格坐标。

三、代码实现

class Solution {
public:
    vector<pair<int, int> > pacificAtlantic(vector<vector<int> >& matrix) {
        vector<pair<int,int> >ans;
        int rows=matrix.size();
        if(!rows) return ans;
        int cols=matrix[0].size();
        if(!cols) return ans;
        vector<vector<bool> >statusP(rows,vector<bool>(cols,false));
        vector<vector<bool> >statusA(rows,vector<bool>(cols,false));
        queue<pair<int,int> >qP;//用于DFS能够流到太平洋的点
        queue<pair<int,int> >qA;//用于DFS能够流到大西洋的点

        //初始化
        for(int i=0;i<rows;i++){
            statusP[i][0]=true;
            statusA[i][cols-1]=true;
            qP.push(pair<int,int>(i,0));
            qA.push(pair<int,int>(i,cols-1));
        }
        for(int i=0;i<cols;i++){
            statusP[0][i]=true;
            statusA[rows-1][i]=true;
            qP.push(pair<int,int>(0,i));
            qA.push(pair<int,int>(rows-1,i));
        }
        int dx[4]={-1,1,0,0};
        int dy[4]={0,0,-1,1};

        //广度优先搜索能够流到太平洋的点
        while(!qP.empty()){
            pair<int,int> cur=qP.front();
            qP.pop();
            int cur_x=cur.first;
            int cur_y=cur.second;
            for(int i=0;i<4;i++){
                int nx=cur_x+dx[i];
                int ny=cur_y+dy[i];
                if(nx>=0 && nx<rows && ny>=0 && ny<cols && !statusP[nx][ny] && matrix[cur_x][cur_y]<=matrix[nx][ny]){
                    statusP[nx][ny]=true;
                    qP.push(pair<int,int>(nx,ny));
                }
            }
        }

        //广度优先搜索能够流到大西洋的点
        while(!qA.empty()){
            pair<int,int> cur=qA.front();
            qA.pop();
            int cur_x=cur.first;
            int cur_y=cur.second;
            for(int i=0;i<4;i++){
                int nx=cur_x+dx[i];
                int ny=cur_y+dy[i];
                if(nx>=0 && nx<rows && ny>=0 && ny<cols && !statusA[nx][ny] && matrix[cur_x][cur_y]<=matrix[nx][ny]){
                    statusA[nx][ny]=true;
                    qA.push(pair<int,int>(nx,ny));
                }
            }
        }

        //寻找既能流到太平洋又能流到大西洋的点
        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                if(statusP[i][j] && statusA[i][j]){
                    ans.push_back(pair<int,int>(i,j));
                }
            }
        }
        return ans;
    }
};

四、参考资料

无