题目描述
题目链接:https://leetcode-cn.com/problems/swim-in-rising-water/
解题思路
和778. 水位上升的泳池中游泳差不多模子,思路都是二维表每个格子当作图节点,一个图中找路径的问题。
数组范围确定了的。
第一种解决办法:并查集
给平台每个格子编号,从左到右从上到下0~N*N-1,把平台高度当作key,平台编号当作value存到一个数组,这一步避免排序!
核心思想:一个i增长表示此时水位高度,平台高度低于这个水位高度的就并入此时连通子图中,其实就是看一下相邻四周平台的高度是否小于当前平台高度,小于就合并。
每次合并检查一下0号节点和N*N-1号节点是否联通,联通就结束。
由于平台高度<N,因此i增长有界限。
第二种解决办法:二分+搜索
N的范围50,解的范围也确定0~N-1,不断尝试范围内的解,找最小的解,搜索可选择BFS或者DFS复杂度NN,找解的时候用二分搜索。最终复杂度O(N^2logN)
并查集解决Java实现
class Solution {
static int []p = new int[2505];
static int[][]Direction={{0,-1},{0,1},{-1,0},{1,0}};
public int swimInWater(int[][] grid) {
int N = grid.length;
int len = N*N;
int []h = new int[N*N];
for(int i = 0;i < N;i++){
for(int j = 0;j < N;j++){
h[grid[i][j]] = i*N+j;
p[grid[i][j]] = grid[i][j];
}
}
int ans = 0;
for(int i = 0;i < len;i++){
int x = h[i]/N;
int y = h[i]%N;
for(int j=0;j<4;j++){
int NewX = x+Direction[j][0];
int NewY = y+Direction[j][1];
if(Check(NewX,NewY,N)&&grid[NewX][NewY]<grid[x][y]){
Union(h[i],NewX*N+NewY);
}
}
if(Find(0)==Find(len-1)){
ans = i;
break;
}
}
return ans;
}
static void Union(int x,int y){
int px = Find(x);
int py = Find(y);
if(px!=py){
p[px] = py;
}
}
static int Find(int x){
return x==p[x]?x:(p[x]=Find(p[x]));
}
static boolean Check(int x,int y,int N){
return x>=0&&x<N&&y>=0&&y<N;
}
}