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;
    }
};
posted on 2021-06-06 18:40  雾恋过往  阅读(47)  评论(0编辑  收藏  举报

Live2D