题目大意就是求出,能同时流向大西洋和太平洋的坐标集合
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
超时。。。