题目
有一个 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]]
解题
方法一:逆向dfs+状态压缩(节约空间)
从太平洋、大西洋的边缘出发,朝高处dfs
如果是从太平洋出发的标记为1
如果是从大西洋出发的标记为2
由int的不同位,来标记,实现状态压缩,节约空间
对于最终标记结果为3(从太平洋出发,又是从大西洋出发)的就是想要的点
class Solution { public: vector<vector<int>> dirs={{-1,0},{0,-1},{1,0},{0,1}}; vector<vector<char>> matrix; int m,n; void dfs(vector<vector<int>>& heights,int x,int y,bool flag){ for(vector<int>& dir:dirs){ int nx=x+dir[0]; int ny=y+dir[1]; if(nx<0||nx>=m||ny<0||ny>=n||heights[nx][ny]<heights[x][y]) continue; if(flag&&(matrix[nx][ny]&2)) continue; if(!flag&&(matrix[nx][ny]&1)) continue; if(flag){ matrix[nx][ny]|=2; }else{ matrix[nx][ny]|=1; } dfs(heights,nx,ny,flag); } } vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) { m=heights.size(),n=heights[0].size(); matrix=vector<vector<char>>(m,vector<char>(n,0)); for(int i=0;i<m;i++){ for(int j=0;j<n;j++){ if(i==0||j==0){//太平洋出发 matrix[i][j]|=1; dfs(heights,i,j,false); } if(i==m-1||j==n-1){//大西洋出发 matrix[i][j]|=2; dfs(heights,i,j,true); } } } vector<vector<int>> res; for(int i=0;i<m;i++){ for(int j=0;j<n;j++){ if(matrix[i][j]==3) res.push_back({i,j}); } } return res; } };