一、Problem
在一个 N x N 的坐标方格 grid 中,每一个方格的值 grid[i][j] 表示在位置 (i,j) 的平台高度。
现在开始下雨了。当时间为 t 时,此时雨水导致水池中任意位置的水位为 t 。你可以从一个平台游向四周相邻的任意一个平台,但是前提是此时水位必须同时淹没这两个平台。假定你可以瞬间移动无限距离,也就是默认在方格内部游动是不耗时的。当然,在你游泳的时候你必须待在坐标方格里面。
你从坐标方格的左上平台 (0,0) 出发。最少耗时多久你才能到达坐标方格的右下平台 (N-1, N-1)?
输入: [[0,1,2,3,4],[24,23,22,21,5],[12,13,14,15,16],[11,17,18,19,20],[10,9,8,7,6]]
输出: 16
解释:
0 1 2 3 4
24 23 22 21 5
12 13 14 15 16
11 17 18 19 20
10 9 8 7 6
最终的路线用加粗进行了标记。
我们必须等到时间为 16,此时才能保证平台 (0, 0) 和 (4, 4) 是连通的
提示:
2 <= N <= 50.
grid[i][j] 位于区间 [0, …, N*N - 1] 内。
二、Solution
方法一:二分+bfs
直接秒杀…,二分 + dfs 也可以,本质是二分时间 t,在加一个起点和终点连通性检查
const int N=55;
class Solution {
public:
struct node {int x, y;};
int n, m, vis[N][N], d[4][2] = { {1,0},{0,-1},{0,1},{-1,0} };
bool bfs(queue<node>& q, int& t, vector<vector<int>>& g) {
while (!q.empty()) for (int i=q.size(); i; i--) {
node now=q.front(); q.pop();
if (now.x==n-1 && now.y==m-1)
return true;
for (int k=0; k<4; k++) {
int tx=now.x+d[k][0], ty=now.y+d[k][1];
if (tx>=0 && tx<n && ty>=0 && ty<m && !vis[tx][ty] && g[tx][ty] <= t) {
q.push({tx, ty});
vis[tx][ty]=1;
}
}
}
return false;
}
int swimInWater(vector<vector<int>>& g) {
n=g.size(), m=g[0].size();
int l=0, r=3000;
while (l < r) {
int t = l + (r-l) / 2;
if (g[0][0] > t) {
l=t+1; continue;
}
queue<node> q;
q.push({0, 0});
memset(vis, 0, sizeof vis);
vis[0][0]=1;
if (bfs(q, t, g)) r = t;
else l = t+1;
}
return l;
}
};
复杂度分析
- 时间复杂度: O ( n l o g n ) O(nlogn) O(nlogn),
- 空间复杂度: O ( n 2 ) O(n^2) O(n2)