你要开发一座金矿,地质勘测学家已经探明了这座金矿中的资源分布,并用大小为 m * n
的网格 grid
进行了标注。每个单元格中的整数就表示这一单元格中的黄金数量;如果该单元格是空的,那么就是 0
。
为了使收益最大化,矿工需要按以下规则来开采黄金:
0
的单元格。示例 1:
输入:grid = [[0,6,0],[5,8,7],[0,9,0]]
输出:24
解释:
[[0,6,0],
[5,8,7],
[0,9,0]]
一种收集最多黄金的路线是:9 -> 8 -> 7。
示例 2:
输入:grid = [[1,0,7],[2,0,6],[3,4,5],[0,3,0],[9,0,20]]
输出:28
解释:
[[1,0,7],
[2,0,6],
[3,4,5],
[0,3,0],
[9,0,20]]
一种收集最多黄金的路线是:1 -> 2 -> 3 -> 4 -> 5 -> 6 -> 7。
提示:
1 <= grid.length, grid[i].length <= 15
0 <= grid[i][j] <= 100
解题思路
典型的dfs
问题,通过遍历grid
,判断当期的值是不是大于0
,如果是的话,那么我们可以开始挖掘。接着思路就是dfs
了,遍历的过程中,对于遍历过的区域将它标记为0
即可。
class Solution:
def getMaximumGold(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid[0])
d = [(0, 1), (1, 0), (0, -1), (-1, 0)]
def dfs(x, y):
res = 0
src = grid[x][y]
grid[x][y] = 0
for i, j in d:
nx = i + x
ny = j + y
if nx >= 0 and nx < m and ny >= 0 and ny < n and grid[nx][ny] != 0:
res = max(res, grid[nx][ny] + dfs(nx, ny))
grid[x][y] = src
return res
res = 0
for i in range(m):
for j in range(n):
if grid[i][j] != 0:
res = max(res, grid[i][j] + dfs(i, j))
return res
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!