Leetcode每日一题0130——778. 水位上升的泳池中游泳（Python）

题目内容

在一个 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
解释:
 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] 的排列。

解题思考

与前几日所刷蓝桥杯最短路问题使用的SPFA算法link.类似

算法维护一个队列queue与结果矩阵ans，每次对队首位置(x,y)的四近邻(px,py)搜索，更新ans[px][py]为到达当前位置(px,py)所需最短时间。

代码

class Solution(object):
    def swimInWater(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        n = len(grid)
        queue = [(0,0)]
        ans = [[float("inf") for _ in range(n)] for __ in range(n)]
        ans[0][0] = grid[0][0]

        while queue:
            x, y = queue[0]
            queue.remove(queue[0])
            for dx, dy in [ (0,1), (1,0), (-1,0), (0,-1) ]:
                px, py = dx+x, dy+y
                if 0 <= px < n and 0 <= py < n:
                    if max(grid[px][py],ans[x][y]) < ans[px][py]:
                        ans[px][py] = max(grid[px][py],ans[x][y])
                        queue.append((px, py))

        return ans[-1][-1]