Skip to content

Commit b844ace

Browse files
committed
[LeetCode Sync] Runtime - 31 ms (54.22%), Memory - 18.7 MB (12.83%)
1 parent 4e1061a commit b844ace

File tree

2 files changed

+68
-0
lines changed

2 files changed

+68
-0
lines changed
Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
<p>You are given an <code>n x n</code> integer matrix <code>grid</code> where each value <code>grid[i][j]</code> represents the elevation at that point <code>(i, j)</code>.</p>
2+
3+
<p>It starts raining, and water gradually rises over time. At time <code>t</code>, the water level is <code>t</code>, meaning <strong>any</strong> cell with elevation less than equal to <code>t</code> is submerged or reachable.</p>
4+
5+
<p>You can swim from a square to another 4-directionally adjacent square if and only if the elevation of both squares individually are at most <code>t</code>. You can swim infinite distances in zero time. Of course, you must stay within the boundaries of the grid during your swim.</p>
6+
7+
<p>Return <em>the minimum time until you can reach the bottom right square </em><code>(n - 1, n - 1)</code><em> if you start at the top left square </em><code>(0, 0)</code>.</p>
8+
9+
<p>&nbsp;</p>
10+
<p><strong class="example">Example 1:</strong></p>
11+
<img alt="" src="https://assets.leetcode.com/uploads/2021/06/29/swim1-grid.jpg" style="width: 164px; height: 165px;" />
12+
<pre>
13+
<strong>Input:</strong> grid = [[0,2],[1,3]]
14+
<strong>Output:</strong> 3
15+
Explanation:
16+
At time 0, you are in grid location (0, 0).
17+
You cannot go anywhere else because 4-directionally adjacent neighbors have a higher elevation than t = 0.
18+
You cannot reach point (1, 1) until time 3.
19+
When the depth of water is 3, we can swim anywhere inside the grid.
20+
</pre>
21+
22+
<p><strong class="example">Example 2:</strong></p>
23+
<img alt="" src="https://assets.leetcode.com/uploads/2021/06/29/swim2-grid-1.jpg" style="width: 404px; height: 405px;" />
24+
<pre>
25+
<strong>Input:</strong> grid = [[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]]
26+
<strong>Output:</strong> 16
27+
<strong>Explanation:</strong> The final route is shown.
28+
We need to wait until time 16 so that (0, 0) and (4, 4) are connected.
29+
</pre>
30+
31+
<p>&nbsp;</p>
32+
<p><strong>Constraints:</strong></p>
33+
34+
<ul>
35+
<li><code>n == grid.length</code></li>
36+
<li><code>n == grid[i].length</code></li>
37+
<li><code>1 &lt;= n &lt;= 50</code></li>
38+
<li><code>0 &lt;= grid[i][j] &lt;&nbsp;n<sup>2</sup></code></li>
39+
<li>Each value <code>grid[i][j]</code> is <strong>unique</strong>.</li>
40+
</ul>
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
class Solution:
2+
def swimInWater(self, grid: List[List[int]]) -> int:
3+
n = len(grid)
4+
m = n ** 2
5+
p = list(range(m))
6+
high = [0] * m
7+
8+
def find(x: int) -> int:
9+
if p[x] != x:
10+
p[x] = find(p[x])
11+
return p[x]
12+
13+
for i, row in enumerate(grid):
14+
for j, h in enumerate(row):
15+
high[h] = i * n + j
16+
17+
directions = (-1, 0, 1, 0, -1)
18+
for t in range(m):
19+
div, remain = divmod(high[t], n)
20+
for dx, dy in pairwise(directions):
21+
nx, ny = div + dx, remain + dy
22+
if 0 <= nx < n and 0 <= ny < n and grid[nx][ny] <= t:
23+
p[find(div * n + remain)] = find(nx * n + ny)
24+
25+
if find(0) == find(m - 1):
26+
return t
27+
28+
return 0

0 commit comments

Comments
 (0)