礼物的最大价值

题目描述

来源于 https://leetcode-cn.com/

在一个 m*n 的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?

示例 1:

输入: 
[
  [1,3,1],
  [1,5,1],
  [4,2,1]
]
输出: 12
解释: 路径 1→3→5→2→1 可以拿到最多价值的礼物

提示:

  • 0 < grid.length <= 200
  • 0 < grid[0].length <= 200

解法:

用动态规划,计算在每个位置上可以得到的礼物价值总和的最大值。

class Solution {
public:
    int maxValue(vector<vector<int>>& grid) {
        int n_rows = grid.size();
        int n_cols = grid[0].size();
        for(int row=0;row<n_rows;row++){
            for(int col=0;col<n_cols;col++){
                int n = 0;
                if(row-1 >= 0){
                    n = grid[row-1][col];
                }
                if(col-1 >= 0){
                    n = max(n, grid[row][col-1]);
                }
                grid[row][col] += n;
            }
        }
        return grid.back().back();
    }
};

上面这种方法,使用原矩阵作为动态规划的 dp 矩阵,这会修改输入数据,如果要求不修改输入。那就需要创建一个 dp 矩阵。上面的方法中 dp 的大小是 m*n。但其实可以使用一维的向量来存储中间值。

使用一个向量 dp,存储在每一列上所能得到的最大值。因为在每个点上,移动方向只能是向右或者向下。如果向右,那么 dp[i] = dp[i-1] + grid[row][i]。如果向下移动,那么 dp[i] = dp[i] + grid[row][i]

所以在每个点上 (row, i),只需要做如下更新 dp[i] = max(dp[i], dp[i-1]) + grid[row][i]

class Solution {
public:
    int maxValue(vector<vector<int>>& grid) {
        int n_rows = grid.size();
        int n_cols = grid[0].size();
        vector<int> dp(n_cols, 0);
        for(int row=0;row<n_rows;row++){
            for(int col=0;col<n_cols;col++){
                int n = 0;
                if(col > 0){
                    n = max(dp[col], dp[col-1]);
                }else{
                    n = dp[col];
                }
                dp[col] = n + grid[row][col];
            }
        }
        return dp.back();
    }
};