LeetCode1219 黄金矿工
题目描述:
你要开发一座金矿,地质勘测学家已经探明了这座金矿中的资源分布,并用大小为 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。
输入: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。
class Solution {
private:
vector<vector<int>> g;
int result=0;
public:
int max(int a,int b){
return a>b?a:b;
}
bool canMove(int x, int y) { // 判断是否这个点是否可以走
return (x >= 0 && x < g.size()) && (y >= 0 && y < g[0].size()) && g[x][y];
}
void dfs(int i,int j,int cur){
cur += g[i][j]; // 拿到该点的黄金
result = max(result, cur); // 记录最大值
int goldValue=g[i][j]; //保存当前点的值
g[i][j]=0; //并将grid[i][j]置为0 避免在下一次递归的时候计算回来
//分上下左右四个方向搜索
if(canMove(i-1,j)) dfs(i-1,j,cur);
if(canMove(i+1,j)) dfs(i+1,j,cur);
if(canMove(i,j-1)) dfs(i,j-1,cur);
if(canMove(i,j+1)) dfs(i,j+1,cur);
g[i][j]=goldValue;
}
int getMaximumGold(vector<vector<int>>& grid) {
g=grid;
for(int i=0;i<grid.size();++i){
for(int j=0;j<grid[0].size();++j){
if(grid[i][j]){
dfs(i,j,0);
}
}
}
return result;
}
};