每日随机一题 leetcode778. 水位上升的泳池中游泳

题:
在这里插入图片描述


思:
使用迪杰斯特拉算法的思想,对其进行变化,改为:每次选一个当前可选的最小的边,当右下角那个点被遍历时,结束程序。
刚开始只有(0,1)和(1,0)位置的点是可以选的,于是将它们加入到待遍历的点集合里。
每次从待遍历的边里选一个权值最小的点,并更新当前最小值。然后将其上下左右还没有加入过队列的点,加入到队列中。
当右下角那个点被遍历的时候,代表存在了一条连通的路使得左上角到右下角连通,并且此时最小值即为答案。
使用优先队列来存放待遍历的点。


码:

class Node {
        int x;
        int y;
        int weight;

        Node(int x, int y, int weight) {
            this.x = x;
            this.y = y;
            this.weight = weight;
        }
    }

    public int swimInWater(int[][] grid) {
        // 使用迪杰斯特拉算法的思路,每次选一个最小边,当右下角那个点被遍历时,结束程序
        int min = grid[0][0];
        // 待遍历的集合
        Queue<Node> priOrityQueue = new PriorityQueue<Node>(new Comparator<Node>() {
            @Override
            public int compare(Node o1, Node o2) {
                return o1.weight - o2.weight;
            }
        });
        int N = grid.length;
        // 已遍历的集合(指已经放在队列里)
        int valid[][] = new int[N][N];
        valid[0][0] = 1;
        priOrityQueue.offer(new Node(0, 1, grid[0][1]));
        priOrityQueue.offer(new Node(1, 0, grid[1][0]));
        valid[0][1] = 1;
        valid[1][0] = 1;
        // 初始化完毕

        while (!priOrityQueue.isEmpty()) {
            Node node = priOrityQueue.poll();
            min = Math.max(min, node.weight);
            if (node.x == N - 1 && node.y == N - 1)
                break;
            if (node.x - 1 >= 0 && valid[node.x - 1][node.y] == 0) {
                priOrityQueue.offer(new Node(node.x - 1, node.y, grid[node.x - 1][node.y]));
                valid[node.x - 1][node.y] = 1;
            }
            if (node.x + 1 < N && valid[node.x + 1][node.y] == 0) {
                priOrityQueue.offer(new Node(node.x + 1, node.y, grid[node.x + 1][node.y]));
                valid[node.x + 1][node.y] = 1;
            }
            if (node.y - 1 >= 0 && valid[node.x][node.y - 1] == 0) {
                priOrityQueue.offer(new Node(node.x, node.y - 1, grid[node.x][node.y - 1]));
                valid[node.x][node.y - 1] = 1;
            }
            if (node.y + 1 < N && valid[node.x][node.y + 1] == 0) {
                priOrityQueue.offer(new Node(node.x, node.y + 1, grid[node.x][node.y + 1]));
                valid[node.x][node.y + 1] = 1;
            }
        }
        return min;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值