leetcode 1219. 黄金矿工（C++）

你要开发一座金矿，地质勘测学家已经探明了这座金矿中的资源分布，并用大小为 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 个单元格中有黄金。

C++

class Solution {
public:
    void dfs(vector<vector<int>>& grid, vector<vector<int>>& flag, int r, int c, int tmp, int* ans)
    {
        int dx[4]={0,0,1,-1};
        int dy[4]={1,-1,0,0};
        if(grid[r][c] && 0==flag[r][c])
        {
            flag[r][c]=1;
            tmp+=grid[r][c];
            *ans=max(*ans,tmp);
            for(int k=0;k<4;k++)
            {
                int y=r+dy[k];
                int x=c+dx[k];
                if(y>=0 && y<grid.size() && x>=0 && x<grid[0].size())
                {
                    dfs(grid,flag,y,x,tmp,ans);
                }
            }
            flag[r][c]=0;
        }
        return;
    }
    
    int getMaximumGold(vector<vector<int>>& grid) 
    {
        int ans=0;
        int tmp=0;
        int m=grid.size();
        int n=grid[0].size();
        vector<vector<int>> flag(m,vector<int>(n,0));
        int* q=&ans;
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                dfs(grid,flag,i,j,tmp,q);
            }
        }
        return ans;
    }
};