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

这种题目有好几种写法
1、裸BFS,有点像SPFA

class Solution {
public:
    int swimInWater(vector<vector<int>>& heights) {
        int n = heights.size(), m = heights[0].size();
        vector<vector<int>> f(n, vector<int>(m, INT_MAX));
        int dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};

        f[0][0] = heights[0][0];
        queue<int> q;
        q.push(0);
        while(!q.empty()) {
            auto t = q.front();
            q.pop();
            int x = t / m;
            int y = t % m;

            for (int d = 0; d < 4; d++) {
                int a = x + dx[d], b = y + dy[d];
                if (a >= 0 && a < n && b >= 0 && b < m) {
                    int tmp = max(f[x][y], heights[a][b]);
                    if (tmp >= f[a][b]) continue;

                    f[a][b] = tmp;
                    q.push(a*m + b);
                }
            }
        }
        return f[n-1][m-1];
    }
};	

2、并查集
并查集的思路挺奇妙的,首先我们从小到大枚举时间,如果说 时间 大于矩阵的某一个格子的高度,如果同时也大于格子上下左右的格子高度,那么俩个格子就是联通的,就可以加一条边了,每个格子最多加四条边。如果说第一个格子和最后一个格子联通了,那么算法就可以结束了。

class Solution {
public:
    int pre[2500+7];
    int dir[4][2]={{0,1},{1,0},{0,-1},{-1,0}};
    int find(int x){
        return pre[x]==x ? x:pre[x]=find(pre[x]);
    }
    void merge(int x,int y){
        int fx=find(x);
        int fy=find(y);
        if(fx!=fy) pre[fx]=fy;
    }
    vector<int >ve[2500+7];
    int swimInWater(vector<vector<int>>& grid) {
        int n = grid.size();
        int m = grid[0].size();
        int end = n*m-1;
        for(int i=0;i<=end;i++){
            ve[grid[i/m][i%m]].push_back(i);
        }
        for(int i=0;i<=end;i++) pre[i]=i;
        for(int ans=0;ans<=end;ans++){
            for(int j=0;j<ve[ans].size();j++){
                int i=ve[ans][j];
                int tx=i/m;
                int ty=i%m;
                if(ans<grid[tx][ty]) continue ;
                for(int j=0;j<4;j++){
                    int dx=tx+dir[j][0];
                    int dy=ty+dir[j][1];
                    if(dx<0||dy<0||dx>=n||dy>=m) continue ;
                    if(grid[dx][dy]>ans) continue ;
                    merge(dx*m+dy,i);
                }
            }
            if(find(0)==find(n*m-1)) return ans;
        }
        return -1;
    }
};

3、最短路(dijkstra)
待补
4、二分+bfs
可以二分一个时间,然后用bfs判断能否到达终点。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值