leetcode 417. Pacific Atlantic Water Flow

题目大意就是求出，能同时流向大西洋和太平洋的坐标集合

int vis[4][2] = {0,1,1,0,0,-1,-1,0};
    vector<pair<int,int>> np;
    bool ret1[160][160],ret2[160][160];
    int see1[160][160],see2[160][160];

    void dfs(int see[][160],vector<vector<int> >& matrix,int a,int b,bool ai) {
        see[a][b] = 1;

        if(ai) ret1[a][b] = 1;
        else ret2[a][b] = 1;

        if(ret1[a][b] && ret2[a][b]) np.push_back(make_pair(a,b));

        for(int i = 0;i < 4;++i){
            int nx = a + vis[i][0];
            int ny = b + vis[i][1];
            if(nx >= 0 && ny >= 0 && nx < matrix.size() && ny < matrix[0].size() && matrix[nx][ny] >= matrix[a][b] && !see[nx][ny]) {
                dfs(see,matrix,nx,ny,ai);
            }
        }
    }

    void bfs(int see[][160],vector<vector<int> >& matrix,int a,int b,bool ai) {
        queue<vector<int>>q;
        q.push({a,b});
        while(!q.empty()) {
            int x = q.front()[0];
            int y = q.front()[1];
            q.pop();
            see[x][y] = 1;

            if(ai) ret1[x][y] = 1;
            else ret2[x][y] = 1;

            if(ret1[x][y] && ret2[x][y]) np.push_back(make_pair(x,y));

            for(int i = 0;i < 4;++i){
                int nx = x + vis[i][0];
                int ny = y + vis[i][1];
                if(nx >= 0 && ny >= 0 && nx < matrix.size() && ny < matrix[0].size() && matrix[nx][ny] >= matrix[x][y] && !see[nx][ny]) {
                    q.push({nx,ny});
                }
            }
        }
    }

    vector<pair<int, int> > pacificAtlantic(vector<vector<int>>& matrix) {
        if(!matrix.size()) return np;

        for(int i = 0;i < matrix.size();++i) {
            dfs(see1,matrix,i,0,1);
            dfs(see2,matrix,i,matrix[0].size()-1,0);
        }
        for(int i = 0;i < matrix[0].size();++i) {
            dfs(see1,matrix,0,i,1);
            dfs(see2,matrix,matrix.size()-1,i,0);
        }
        sort(np.begin(),np.end());
        np.erase(unique(np.begin(),np.end()),np.end());
        return np;
    }

顺便说一句，dfs 47ms AC,而bfs

超时。。。