【算法题解】 Day24 动态规划

85 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

⭐ 算法题解系列文章旨在精选重点与易错的算法题,总结常见的算法思路与可能出现的错误,以实战习题的形式理解算法,使用算法。

🔥 本文已收录于算法刷题系列专栏: 算法题解 欢迎订阅,持续更新。

🎉 欢迎关注👀 点赞👍 收藏⭐ 留言📝

💬 代码成就万世基,积沙镇海;梦想永在凌云意,意气风发; 𝓼𝓲𝓭𝓲𝓸𝓽  

剑指 Offer 42. 连续子数组的最大和

题目

剑指 Offer 42. 连续子数组的最大和 难度:easy

输入一个整型数组,数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。

要求时间复杂度为 O(n)O(n)

示例1:

输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。

提示:

  • 1 <= arr.length <= 10^5
  • -100 <= arr[i] <= 100  

方法一:动态规划

思路

假设 nums\textit{nums} 数组的长度是 n,下标从 0 到 n−1。

我们用 f(i)f(i) 代表以第 i 个数结尾的「连续子数组的最大和」,那么很显然我们要求的答案就是:

max0in1{f(i)}\max_{0 \leq i \leq n-1} \{ f(i) \}

因此我们只需要求出每个位置的 f(i)f(i),然后返回 f 数组中的最大值即可。那么我们如何求 f(i)f(i)呢?我们可以考虑 nums[i]\textit{nums}[i] 单独成为一段还是加入 f(i1)f(i-1) 对应的那一段,这取决于 nums[i]\textit{nums}[i]f(i1)+nums[i]f(i-1) + \textit{nums}[i] 的大小,我们希望获得一个比较大的,于是可以写出这样的动态规划转移方程:

f(i)=max{f(i1)+nums[i],nums[i]}f(i) = \max \{ f(i-1) + \textit{nums}[i], \textit{nums}[i] \}

不难给出一个时间复杂度 O(n)O(n)、空间复杂度 O(n)O(n) 的实现,即用一个 ff 数组来保存 f(i)f(i) 的值,用一个循环求出所有 f(i)f(i)。考虑到 f(i)f(i) 只和 f(i1)f(i-1) 相关,于是我们可以只用一个变量 pre\textit{pre} 来维护对于当前 f(i)f(i)f(i1)f(i-1) 的值是多少,从而让空间复杂度降低到 O(1)O(1),这有点类似「滚动数组」的思想。  

解题

Python:

class Solution:
    def maxSubArray(self, nums: List[int]) -> int:
        for i in range(1, len(nums)):
            nums[i] += max(nums[i - 1], 0)
        return max(nums)

Java:

class Solution {
    public int maxSubArray(int[] nums) {
        int pre = 0, maxAns = nums[0];
        for (int x : nums) {
            pre = Math.max(pre + x, x);
            maxAns = Math.max(maxAns, pre);
        }
        return maxAns;
    }
}

 

剑指 Offer 47. 礼物的最大价值

题目

剑指 Offer 47. 礼物的最大价值 难度:medium

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

示例 1:

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

提示:

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

方法一:动态规划

思路

根据题目说明,易得某单元格只可能从上边单元格或左边单元格到达。

f(i,j)f(i, j) 为从棋盘左上角走至单元格 (i,j)(i ,j) 的礼物最大累计价值,易得到以下递推关系:f(i,j)f(i,j) 等于 f(i,j1)f(i,j-1)f(i1,j)f(i-1,j) 中的较大值加上当前单元格礼物价值 grid(i,j)grid(i,j)

f(i,j)=max[f(i,j1),f(i1,j)]+grid(i,j)f(i,j) = \max[f(i,j-1), f(i-1,j)] + grid(i,j)

因此,可用动态规划解决此问题,以上公式便为转移方程。

image.png  

解题

Python:

class Solution:
    def maxValue(self, grid: List[List[int]]) -> int:
        m, n = len(grid), len(grid[0])
        for j in range(1, n): # 初始化第一行
            grid[0][j] += grid[0][j - 1]
        for i in range(1, m): # 初始化第一列
            grid[i][0] += grid[i - 1][0]
        for i in range(1, m):
            for j in range(1, n):
                grid[i][j] += max(grid[i][j - 1], grid[i - 1][j])
        return grid[-1][-1]

Java:

class Solution {
    public int maxValue(int[][] grid) {
        int m = grid.length, n = grid[0].length;
        for(int j = 1; j < n; j++) // 初始化第一行
            grid[0][j] += grid[0][j - 1];
        for(int i = 1; i < m; i++) // 初始化第一列
            grid[i][0] += grid[i - 1][0];
        for(int i = 1; i < m; i++)
            for(int j = 1; j < n; j++) 
                grid[i][j] += Math.max(grid[i][j - 1], grid[i - 1][j]);
        return grid[m - 1][n - 1];
    }
}

 

后记

以上就是 【算法题解】 Day24 动态规划 的所有内容了,创作不易,多多支持 👍👍👍

我是 𝓼𝓲𝓭𝓲𝓸𝓽,期待你的关注 💖💖💖

🔥 系列专栏: 算法题解