首页 > 技术文章 > 礼物的最大值(Python and C++解法)

kongzimengzixiaozhuzi 2020-07-15 21:49 原文

题目:

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

示例 1:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/li-wu-de-zui-da-jie-zhi-lcof

思路:

  使用动态规划解决。

  状态定义:设动态规划矩阵 maxStore,maxStore[i][j]代表从棋盘的左上角开始,到达单元格(i,j)时能拿到礼物的最大累计价值。

  转移方程:当 i = 0且 j = 0时,为起始元素;

       当 i=0 且 j > ​0 时,为矩阵第一行元素,只可从左边到达;

       当 i > 0 且 j = 0 时,为矩阵第一列元素,只可从上边到达;

       当 i > 0 且 j > 0 时,可从左边或上边到达;

Python解法:

 1 class Solution:
 2     def maxValue(self, grid: List[List[int]]) -> int:
 3         rows, columns = len(grid), len(grid[0])
 4         if rows == 0 or columns == 0:
 5             return 0
 6         maxStore =[[0] * columns] * rows
 7         for i in range(rows):
 8             for j in range(columns):
 9                 if i == 0 and j == 0:  # 起点
10                     maxStore[0][0] = grid[0][0]  
11                 if i == 0 and j > 0:  # 第一行只能从左往右
12                     maxStore[0][j] = maxStore[0][j-1] + grid[0][j]
13                 if j == 0 and i > 0:  # 第一列只能从上往下
14                     maxStore[i][0] = maxStore[i-1][0] + grid[i][0]
15                 if i > 0 and j > 0:
16                     maxStore[i][j] = max(maxStore[i-1][j], maxStore[i][j-1]) + grid[i][j] 
17         return maxStore[rows-1][columns-1]

C++解法:

 1 class Solution {
 2 public:
 3     int maxValue(vector<vector<int>>& grid) {
 4         int rows = grid.size(), columns = grid[0].size();
 5         if(rows == 0 && columns == 0)
 6             return 0;
 7         // 二维vector初始化方法!!!!!!!!
 8         vector<vector<int>> maxStore(rows, vector<int>(columns, 0));  
 9         for(int i = 0; i< rows; i++) {
10             for(int j = 0; j < columns; j++) {
11                 if(i == 0 && j == 0)
12                     maxStore[0][0] = grid[0][0];
13                 if(i == 0 && j > 0)
14                     maxStore[0][j] = maxStore[0][j-1] + grid[0][j];
15                 if(i > 0 && j == 0)
16                     maxStore[i][0] = maxStore[i-1][0] + grid[i][0];
17                 if(i > 0 && j > 0)
18                     maxStore[i][j] = max(maxStore[i-1][j], maxStore[i][j-1]) + grid[i][j];
19             }
20         }
21         return maxStore[rows-1][columns-1];
22     }
23 };

推荐阅读