题目来源:https://leetcode-cn.com/problems/path-with-maximum-gold/
大致题意:
给定一个矩阵表示图,每个元素值表示对应节点的金矿含量。假设矿工从一个有金矿的节点开始挖,每次只能移动到上下左右其中一个有金矿的位置,求出矿工最多挖出多少金矿
思路
可以理解为用 dfs 求最大金矿
dfs
- 遍历图,以每个含有金矿的节点为起点进行 dfs
- dfs 四个方向中还未遍历过的金矿节点,统计本轮金矿含量,最终答案取最大值
class Solution {
static int[][] dirs = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
int[][] grid;
int m, n;
int ans = 0;
public int getMaximumGold(int[][] grid) {
this.grid = grid;
this.m = grid.length;
this.n = grid[0].length;
// 遍历所有节点
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
// 以金矿节点为起点,进行 dfs
if (grid[i][j] != 0) {
dfs(i, j, 0);
}
}
}
return ans;
}
public void dfs(int x, int y, int gold) {
// 累计本次节点金矿
gold += grid[x][y];
// 更新最大值
ans = Math.max(ans, gold);
// 将当前位置标记为 0,避免重复遍历
int rec = grid[x][y];
grid[x][y] = 0;
// 遍历四个方向的金矿节点
for (int d = 0; d < 4; ++d) {
int nx = x + dirs[d][0];
int ny = y + dirs[d][1];
if (nx >= 0 && nx < m && ny >= 0 && ny < n && grid[nx][ny] > 0) {
dfs(nx, ny, gold);
}
}
// 还原
grid[x][y] = rec;
}
}