LeetCode 778 水位上升的泳池中游泳

题目链接

在一个 N x N 的坐标方格 grid 中，每一个方格的值 grid[i][j] 表示在位置 (i,j) 的平台高度。

现在开始下雨了。当时间为 t 时，此时雨水导致水池中任意位置的水位为 t 。你可以从一个平台游向四周相邻的任意一个平台，但是前提是此时水位必须同时淹没这两个平台。假定你可以瞬间移动无限距离，也就是默认在方格内部游动是不耗时的。当然，在你游泳的时候你必须待在坐标方格里面。

你从坐标方格的左上平台 (0，0) 出发。最少耗时多久你才能到达坐标方格的右下平台 (N-1，N-1)？

示例 1:

输入: [[0,2],[1,3]]
输出: 3
解释:
时间为0时，你位于坐标方格的位置为 (0, 0)。
此时你不能游向任意方向，因为四个相邻方向平台的高度都大于当前时间为 0 时的水位。

等时间到达 3 时，你才可以游向平台 (1, 1). 因为此时的水位是 3，坐标方格中的平台没有比水位 3 更高的，所以你可以游向坐标方格中的任意位置

示例 2:

输入: [[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
解释:
我们必须等到时间为 16，此时才能保证平台 (0, 0) 和 (4, 4) 是连通的

二分+BFS~
这题跟 LeetCode 1631 最小体力消耗路径 几乎一模一样，就是二分最少时间，然后从左上角到右下角 BFS 即可，不同的时 BFS 里面的判别条件稍微改了一下而已，即当前水位高度必须大于等于平台高度，AC代码如下：

class Solution {
private:
    int m, n, dir[4][2] = {0, 1, 0, -1, 1, 0, -1, 0};
public:
    bool check(int x, vector<vector<int>> &grid) {
        queue<pair<int, int>> q;
        vector<int> vis(m * n);
        pair<int, int> a, b;
        q.push({0, 0});
        vis[0] = 1;
        while (!q.empty()) {
            a = q.front();
            q.pop();
            if (a.first == m - 1 && a.second == n - 1) {
                return 1;
            }
            for (int i = 0; i < 4; i++) {
                b.first = a.first + dir[i][0];
                b.second = a.second + dir[i][1];
                if (b.first >= 0 && b.first < m && b.second >= 0 && b.second < n && !vis[b.first * n + b.second] &&
                    grid[b.first][b.second] <= x) {
                    vis[b.first * n + b.second] = 1;
                    q.push(b);
                }
            }
        }
        return 0;
    }

    int swimInWater(vector<vector<int>> &grid) {
        m = grid.size(), n = grid[0].size();
        int l = grid[0][0], r = 2500;
        while (l <= r) {
            int mid = l + r >> 1;
            if (check(mid, grid)) r = mid - 1;
            else l = mid + 1;
        }
        return l;
    }
};