题目
417.太平洋大西洋水流问题
题目大意
有一个 m × n
的矩形岛屿,与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界,而 “大西洋” 处于大陆的右边界和下边界。
这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n
的整数矩阵 heights
, heights[r][c]
表示坐标 (r, c)
上单元格 高于海平面的高度 。
岛上雨水较多,如果相邻单元格的高度 小于或等于 当前单元格的高度,雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
返回 网格坐标 result
的 2D列表 ,其中 result[i] = [ri, ci]
表示雨水可以从单元格 (ri, ci)
流向 太平洋和大西洋 。
样例
示例 1:
输入: heights = [[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]]
输出: [[0,4],[1,3],[1,4],[2,2],[3,0],[3,1],[4,0]]
示例 2:
输入: heights = [[2,1],[1,2]]
输出: [[0,0],[0,1],[1,0],[1,1]]
数据规模
提示:
m == heights.length
n == heights[r].length
1 <= m, n <= 200
- 0 < = h e i g h t s [ r ] [ c ] < = 105 0 <= heights[r][c] <= 105 0<=heights[r][c]<=105
思路
这道题题意卡了我很久,感觉现在读题很难读懂(从GPLT比赛就这样了…)
简述一下题意:水可以从当前位置流向四周高度小于等于当前位置的方格,问有哪些方格可流向太平洋&&大西洋 。
那么反着想:太平洋可以反向流经哪些方格,大西洋可以反向流经哪些方格。那么答案就是被太平洋和大西洋同时流经的方格。
使用BFS实现:搜索过程中同样需要记录每个单元格是否可以从太平洋反向到达以及是否可以从大西洋反向到达。反向搜索结束之后,遍历每个网格,如果一个网格既可以从太平洋反向到达也可以从大西洋反向到达,则该网格满足太平洋和大西洋都可以到达,将该网格添加到答案中。
// short int long float double bool char string void
// array vector stack queue auto const operator
// class public private static friend extern
// sizeof new delete return cout cin memset malloc
// relloc size length memset malloc relloc size length
// for while if else switch case continue break system
// endl reverse sort swap substr begin end iterator
// namespace include define NULL nullptr exit equals
// index col row arr err left right ans res vec que sta
// state flag ch str max min default charray std
// maxn minn INT_MAX INT_MIN push_back insert
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int>PII;
typedef pair<int, string>PIS;
const int maxn=1e6+50;//注意修改大小
long long read(){long long x=0,f=1;char c=getchar();while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}while(isdigit(c)){x=x*10+c-'0';c=getchar();}return x*f;}
ll qpow(ll x,ll q,ll Mod){ll ans=1;while(q){if(q&1)ans=ans*x%Mod;q>>=1;x=(x*x)%Mod;}return ans%Mod;}
int dx[4]={0,0,1,-1};
int dy[4]={-1,1,0,0};
class Solution {
int n,m;
void bfs(int x,int y,vector<vector<int>>& vis,vector<vector<int>>& heights){
PII s=make_pair(x,y);
queue<PII>q;
q.push(s);
while(!q.empty()){
PII now=q.front();q.pop();
int x=now.first,y=now.second;
if(vis[x][y]==1)continue;
vis[x][y]=1;
for(int i=0;i<4;i++){
int nex=x+dx[i],ney=y+dy[i];
if(nex>=n||ney>=m||nex<0||ney<0||heights[x][y]>heights[nex][ney])continue;
q.push(make_pair(nex,ney));
}
}
}
public:
vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
n=heights.size(),m=heights[0].size();
vector<vector<int>>A(n,vector<int>(m));
vector<vector<int>>P(n,vector<int>(m));
for(int i=0;i<m;i++){
bfs(0,i,P,heights);
}
for(int i=0;i<n;i++){
bfs(i,0,P,heights);
}
for(int i=0;i<m;i++){
bfs(n-1,i,A,heights);
}
for(int i=0;i<n;i++){
bfs(i,m-1,A,heights);
}
vector<vector<int>>ans;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(A[i][j]&&P[i][j]){
vector<int>a;
a.emplace_back(i);
a.emplace_back(j);
ans.emplace_back(a);
}
}
}
return ans;
}
};