题目地址:
https://leetcode.com/problems/path-with-maximum-gold/
给定一个 m m m行 n n n列的二维数组 A A A,题目保证只含非负整数。可以任选一个非零位置开始走,每次可以上下左右走一步,且不能走到 0 0 0或者走过的位置,问路径和最大是多少。
思路是DFS。枚举每个非 0 0 0位置,看能得到的最大路径和是多少。代码如下:
class Solution {
public:
int getMaximumGold(vector<vector<int>>& g) {
int res = 0, m = g.size(), n = g[0].size();
vector<vector<bool>> vis(m, vector<bool>(n));
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++)
if (g[i][j]) {
res = max(res, dfs(i, j, g, vis));
}
return res;
}
int dfs(int x, int y, vector<vector<int>>& g, vector<vector<bool>>& vis) {
vis[x][y] = true;
int res = g[x][y];
static int d[] = {-1, 0, 1, 0, -1};
for (int _ = 0; _ < 4; _++) {
int nx = x + d[_], ny = y + d[_ + 1];
if (0 <= nx && nx < g.size() && 0 <= ny && ny < g[0].size() &&
g[nx][ny] && !vis[nx][ny])
res = max(res, g[x][y] + dfs(nx, ny, g, vis));
}
vis[x][y] = false;
return res;
}
};
时间复杂度 O ( m n k ) O(mnk) O(mnk), k k k是 A A A中非零数的个数,空间 O ( m n ) O(mn) O(mn)。