LeetCode-2304 网格中的最小路径代价
Table of Contents
LeetCode-2304 网格中的最小路径代价 #
Solution 1 #
本题和经典问题LeetCode-931 下降最小路径和很相似, 不同之处在于多出了节点路径的代价, 这一点不影响动态规划的思路. 记 $dp[i][j]$ 为 $(i, j)$ 出发到达最后一行的最小代价, 则有如下状态转移方程: $$ dp[i][j] = \underset{0\leq k\leq n - 1}{min}(dp[i + 1][j] + grid[i][j] + moveCost[grid[i][j]][k]) $$ 最后 $\underset{0\leq j\leq n - 1}{min}dp[0][j]$ 即为所求最小代价. 代码如下:
class Solution {
public:
int minPathCost(vector<vector<int>>& grid, vector<vector<int>>& moveCost) {
int m = grid.size();
int n = grid[0].size();
vector<vector<int>> dp(m, vector<int>(n, 1000000007));
for (int j = 0; j < n; j++) {
dp[m - 1][j] = grid[m - 1][j];
}
for (int i = m - 2; i >= 0; i--) {
for (int j = 0; j < n; j++) {
for (int k = 0; k < n; k++) {
dp[i][j] = min(dp[i][j], dp[i + 1][k] + grid[i][j] + moveCost[grid[i][j]][k]);
}
}
}
int ans = dp[0][0];
for (int j = 1; j < n; j++) {
ans = min(ans, dp[0][j]);
}
return ans;
}
};