LeetCode 题解工作台
总价值可以被 K 整除的岛屿数目
给你一个 m x n 的矩阵 grid 和一个正整数 k 。一个 岛屿 是由 正 整数(表示陆地)组成的,并且陆地间 四周 连通(水平或垂直)。 一个岛屿的总价值是该岛屿中所有单元格的值之和。 返回总价值可以被 k 整除 的岛屿数量。 示例 1: 输入: grid = [[0,2,1,0,0],[0…
5
题型
5
代码语言
3
相关题
当前训练重点
中等 · 图·搜索
答案摘要
我们定义一个函数 $\textit{dfs}(i, j)$,它从位置 $(i, j)$ 开始进行 DFS 遍历,并且返回该岛屿的总价值。我们将当前位置的值加入总价值,然后将该位置标记为已访问(例如,将其值设为 0)。接着,我们递归地访问四个方向(上、下、左、右)的相邻位置,如果相邻位置的值大于 0,则继续进行 DFS,并将其值加入总价值。最后,我们返回总价值。 在主函数中,我们遍历整个网格,对于每…
Interview AiBoxInterview AiBox 实时 AI 助手,陪你讲清 图·搜索 题型思路
题目描述
给你一个 m x n 的矩阵 grid 和一个正整数 k。一个 岛屿 是由 正 整数(表示陆地)组成的,并且陆地间 四周 连通(水平或垂直)。
一个岛屿的总价值是该岛屿中所有单元格的值之和。
返回总价值可以被 k 整除 的岛屿数量。
示例 1:
输入: grid = [[0,2,1,0,0],[0,5,0,0,5],[0,0,1,0,0],[0,1,4,7,0],[0,2,0,0,8]], k = 5
输出: 2
解释:
网格中包含四个岛屿。蓝色高亮显示的岛屿的总价值可以被 5 整除,而红色高亮显示的岛屿则不能。
示例 2:
输入: grid = [[3,0,3,0], [0,3,0,3], [3,0,3,0]], k = 3
输出: 6
解释:
网格中包含六个岛屿,每个岛屿的总价值都可以被 3 整除。
提示:
m == grid.lengthn == grid[i].length1 <= m, n <= 10001 <= m * n <= 1050 <= grid[i][j] <= 1061 <= k < = 106
解题思路
方法一:DFS
我们定义一个函数 ,它从位置 开始进行 DFS 遍历,并且返回该岛屿的总价值。我们将当前位置的值加入总价值,然后将该位置标记为已访问(例如,将其值设为 0)。接着,我们递归地访问四个方向(上、下、左、右)的相邻位置,如果相邻位置的值大于 0,则继续进行 DFS,并将其值加入总价值。最后,我们返回总价值。
在主函数中,我们遍历整个网格,对于每个未访问的位置 ,如果其值大于 0,则调用 来计算该岛屿的总价值。如果总价值可以被 整除,则将答案加一。
时间复杂度 ,空间复杂度 。其中 和 分别是网格的行数和列数。
class Solution:
def countIslands(self, grid: List[List[int]], k: int) -> int:
def dfs(i: int, j: int) -> int:
s = grid[i][j]
grid[i][j] = 0
for a, b in pairwise(dirs):
x, y = i + a, j + b
if 0 <= x < m and 0 <= y < n and grid[x][y]:
s += dfs(x, y)
return s
m, n = len(grid), len(grid[0])
dirs = (-1, 0, 1, 0, -1)
ans = 0
for i in range(m):
for j in range(n):
if grid[i][j] and dfs(i, j) % k == 0:
ans += 1
return ans
复杂度分析
| 指标 | 值 |
|---|---|
| 时间 | Depends on the final approach |
| 空间 | Depends on the final approach |
面试官常问的追问
外企场景- question_mark
Candidate understands how to identify islands using DFS/BFS.
- question_mark
Candidate calculates the total value of islands correctly.
- question_mark
Candidate can handle large grid sizes efficiently with proper space-time complexity analysis.
常见陷阱
外企场景- error
Forgetting to mark cells as visited, causing redundant calculations.
- error
Not correctly handling islands that touch the grid boundaries.
- error
Incorrectly summing the values of an island, leading to wrong divisibility checks.
进阶变体
外企场景- arrow_right_alt
Using Union-Find to track connected components instead of DFS/BFS.
- arrow_right_alt
Using different grid traversal techniques like spiral or wave traversal.
- arrow_right_alt
Handling negative or zero values in the grid while adjusting divisibility checks.