没看懂。。。先记录下来,后面有时间再回顾
题目:
你要开发一座金矿,地质勘测学家已经探明了这座金矿中的资源分布,并用大小为 m * n 的网格 grid 进行了标注。每个单元格中的整数就表示这一单元格中的黄金数量;如果该单元格是空的,那么就是 0。
为了使收益最大化,矿工需要按以下规则来开采黄金:
- 每当矿工进入一个单元,就会收集该单元格中的所有黄金。
- 矿工每次可以从当前位置向上下左右四个方向走。
- 每个单元格只能被开采(进入)一次。
- 不得开采(进入)黄金数目为 0 的单元格。
- 矿工可以从网格中 任意一个 有黄金的单元格出发或者是停止。
示例:
输入:grid = [[0,6,0],[5,8,7],[0,9,0]]
输出:24
解释:
[[0,6,0],
[5,8,7],
[0,9,0]]
一种收集最多黄金的路线是:9 -> 8 -> 7。
解析:
此题用的是回溯算法,先枚举起点,在这个网格中,若某个位置的值大于0,就可以作为起点;
var dirs = []struct{ x, y int }{{-1, 0}, {1, 0}, {0, -1}, {0, 1}}
func getMaximumGold(grid [][]int) (ans int) {
var dfs func(x, y, gold int)
dfs = func(x, y, gold int) {
gold += grid[x][y]
if gold > ans {
ans = gold
}
rec := grid[x][y]
grid[x][y] = 0
for _, d := range dirs {
nx, ny := x+d.x, y+d.y
if 0 <= nx && nx < len(grid) && 0 <= ny && ny < len(grid[nx]) && grid[nx][ny] > 0 {
dfs(nx, ny, gold)
}
}
grid[x][y] = rec
}
for i, row := range grid {
for j, gold := range row {
if gold > 0 {
dfs(i, j, 0)
}
}
}
return
}