题目
你要开发一座金矿,地质勘测学家已经探明了这座金矿中的资源分布,并用大小为 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
最多 25 个单元格中有黄金。
解题思路
遍历每一个起点,找到从它出发的最大收益,回溯每一条路径,找到能挖到最多黄金的一个。
代码实现
class Solution(object):
def getMaximumGold(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
move=[(0, 1), (1, 0), (0, -1), (-1, 0)]
def dfs(x,y):
#判断是否出界
if x<0 or y<0 or x==m or y==n or not grid[x][y]:
return 0
temp=grid[x][y]
grid[x][y]=0
mx=max(dfs(x+dx,y+dy) for dx,dy in move)
grid[x][y]=temp
return temp+mx
m,n=len(grid),len(grid[0])
#corner=[[0,1],[1,0],[0,n-1],[m-1,0]]
ans=0
for i in range(m):
for j in range(n):
if grid[i][j]!=0:
ans=max(ans,dfs(i,j))
return ans
这个方法最直接,但占空间多,速度慢