每日一题——778. 水位上升的泳池中游泳

最新推荐文章于 2022-05-16 11:57:16 发布

vwmin

最新推荐文章于 2022-05-16 11:57:16 发布

阅读量93

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/weixin_42955230/article/details/113417070

版权

LeetCode 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用并查集、二分查找和最短路径算法来解决水位上升的泳池中游泳的问题。通过建立并查集结构，博主展示了如何判断两个节点是否联通，并给出了详细的代码实现。此外，还提及了二分答案和最短路的解题思路，但表示这部分内容已在此前的题目中讨论过。

摘要由CSDN通过智能技术生成

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

分类：并查集、二分答案、最短路

并查集思路挺简单的。由小到大挨个加，检查下连上了没就行。

class UnionFind{
private:
    vector<int> u;
    vector<int> sz;
public:
    int member;
    UnionFind(int n):member(n), u(n), sz(n, 1){
        iota(u.begin(), u.end(), 0);
    }
    int root(int i){
        return u[i] == i ? i : u[i] = root(u[i]);
    }
    bool connect(int i, int j){
        int ri = root(i), rj = root(j);
        if(ri == rj){
            return false;
        }
        if(sz[ri] < sz[rj]){
            swap(ri, rj);
        }
        u[rj] = ri;
        sz[ri] += sz[rj];
        member--; 
        return true;
    }
    bool isConnected(int i, int j){
        return root(i) == root(j);
    }
};
class Solution {
public:
    struct edge{
        int i, j, v;
        edge(int i, int j, int v):i(i), j(j), v(v){}
        bool operator < (const edge & other) const {
            return v > other.v;
        }
    };
    int d[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
    int swimInWater(vector<vector<int>>& grid) {
        int n = grid.size();
        priority_queue<edge> q;
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                q.push(edge(i, j, grid[i][j]));
            }
        }

        UnionFind u(n*n);
        while(!q.empty()){
            edge cur = q.top(); q.pop();
            for(int i=0; i<4; i++){
                int x = cur.i + d[i][0], y = cur.j + d[i][1];
                if(x>-1 && y>-1 && x<n && y<n && grid[x][y]<cur.v){
                    u.connect(x*n+y, cur.i*n+cur.j);
                }
                if(u.isConnected(0, n*n-1)) return cur.v;
            }
        }
        return n*n-1;
    }
};