二维DP: leetcode 1186、

leetcode 1186:删除一次得到子数组最大和

题目描述:
给你一个整数数组,返回它的某个 非空 子数组(连续元素)在执行一次可选的删除操作后,所能得到的最大元素总和。换句话说,你可以从原数组中选出一个子数组,并可以决定要不要从中删除一个元素(只能删一次哦),(删除后)子数组中至少应当有一个元素,然后该子数组(剩下)的元素总和是所有子数组之中最大的。

注意,删除一个元素后,子数组 不能为空

示例 1:
输入:arr = [1,-2,0,3]
输出:4
解释:我们可以选出 [1, -2, 0, 3],然后删掉 -2,这样得到 [1, 0, 3],和最大。

示例 2:
输入:arr = [1,-2,-2,3]
输出:3
解释:我们直接选出 [3],这就是最大和。

示例 3:
输入:arr = [-1,-1,-1,-1]
输出:-1
解释:最后得到的子数组不能为空,所以我们不能选择 [-1] 并从中删去 -1 来得到 0。
我们应该直接选择 [-1],或者选择 [-1, -1] 再从中删去一个 -1。

提示:
1 <= arr.length <= 105
-104 <= arr[i] <= 104

思路: 二维DP
dp[i][2] 状态定义成二维的,因为可能已经删除过一个,可能还没有删除过
dp[i][0]: 以i为结尾的子数组没有删除过最大和
dp[i][1]:以i为结尾的子数组删除过一次最大和
此题此外还需要注意的是需要保证最后的数组的长度不能是0,所以在初始化的时候做了一点处理,对长度为1(题目中说长度是>= 1的)时进行了求解。
剩下的部分正常进行状态转移

import sys
maxint = sys.maxsize

class Solution:
    def maximumSum(self, arr: List[int]) -> int:
        dp = [[-maxint for _ in range(2)] for _ in range(len(arr))]
        # dp[i][0]到i还没有删除过 dp[i][1]表示已经删除过一次
        if len(arr) == 1:
            return arr[0]
        dp[0][0] = arr[0]
        dp[0][1] = 0
        dp[1][0] = max(arr[0]+arr[1], arr[1])
        dp[1][1] = max(arr[0], arr[1])
        res = max(dp[1][0], dp[1][1])
        for i in range(2, len(arr)):
            dp[i][0] = max(dp[i - 1][0] + arr[i], arr[i])  #接着之前的继续,还是自己开始
            dp[i][1] = max(dp[i - 1][1] + arr[i], dp[i - 1][0]) # 不删除这次的值和之前已经删除的情况相加,或者删除这次的值
            res = max(res, dp[i][0], dp[i][1])
        return res
        

思路参考的题解链接

leetcode 376:摆动序列

题目描述:
如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为 摆动序列 。第一个差(如果存在的话)可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。

例如, [1, 7, 4, 9, 2, 5] 是一个 摆动序列 ,因为差值 (6, -3, 5, -7, 3) 是正负交替出现的。

相反,[1, 4, 7, 2, 5] 和 [1, 7, 4, 5, 5] 不是摆动序列,第一个序列是因为它的前两个差值都是正数,第二个序列是因为它的最后一个差值为零。
子序列 可以通过从原始序列中删除一些(也可以不删除)元素来获得,剩下的元素保持其原始顺序。

给你一个整数数组 nums ,返回 nums 中作为 摆动序列 的 最长子序列的长度 。

示例 1:
输入:nums = [1,7,4,9,2,5]
输出:6
解释:整个序列均为摆动序列,各元素之间的差值为 (6, -3, 5, -7, 3) 。

示例 2:
输入:nums = [1,17,5,10,13,15,10,5,16,8]
输出:7
解释:这个序列包含几个长度为 7 摆动序列。
其中一个是 [1, 17, 10, 13, 10, 16, 8] ,各元素之间的差值为 (16, -7, 3, -3, 6, -8) 。

示例 3:
输入:nums = [1,2,3,4,5,6,7,8,9]
输出:2

提示:
1 <= nums.length <= 1000
0 <= nums[i] <= 1000

思路: DP
状态定义:
dp[i][2]: 表示到i时摆动序列的最大长度,可以有i也可以没有i
dp[i][0]: 表示考虑到i,且最后一部分是上升的时候最大摆动序列长度
dp[i][1]: 表示考虑到i,且最后一部分是下降的时候最大摆动序列长度
在进行状态转移的时候,两者需要调用,因为需要考虑现在的增或者减的情况

import sys
maxint = sys.maxsize

class Solution:
    def wiggleMaxLength(self, nums: List[int]) -> int:
        dp = [[-maxint for _ in range(2)] for _ in range(len(nums))]
        if len(nums) == 1:
            return len(nums)
        dp[0][0], dp[0][1] = 1, 1
        res = 1
        for i in range(1, len(nums)):
            for j in range(i):  # 需要考虑i之前的所有可能的情况 因为可能之前的元素又饿时候是没有算进去的
                if nums[i] - nums[j] > 0:
                    dp[i][0] = max(dp[j][1] + 1, 1, dp[i][0])
                elif nums[i] - nums[j] < 0:
                    dp[i][1] = max(dp[j][0] + 1, 1, dp[i][1])
                elif nums[i] - nums[j] == 0:
                    dp[i][0] = 1
                    dp[i][1] = 1
            res = max(res, dp[i][0], dp[i][1])
        return res

时间复杂度是O(n^ 2)。

进一步想要减小时间复杂度至O(1)。
参考题解思路链接
在这里插入图片描述
在这里插入图片描述
画个图仔细思考一下确实是这样
相应的代码:

class Solution:
    def wiggleMaxLength(self, nums: List[int]) -> int:
        dp = [[0 for _ in range(2)] for _ in range(len(nums))]
        if len(nums) == 1:
            return len(nums)
        res = 0
        for i in range(1, len(nums)):
            if nums[i] - nums[i-1] > 0:
                dp[i][1] = dp[i-1][0] + 1
                dp[i][0] = dp[i - 1][0]
            elif nums[i] - nums[i-1] < 0:
                dp[i][0] = dp[i-1][1] + 1
                dp[i][1] = dp[i-1][1]
            elif nums[i] - nums[i-1] == 0:
                dp[i][0] = dp[i-1][0]
                dp[i][1] = dp[i-1][1]

        res = max(dp[-1][0], dp[-1][1]) + 1
        return res

时间复杂度是O(n),空间复杂度可以进一步优化,将dp数组转换成用两个数进行。

class Solution:
    def wiggleMaxLength(self, nums: List[int]) -> int:
        dp0, dp1 = 0, 0
        if len(nums) == 1:
            return len(nums)
        for i in range(1, len(nums)):
            if nums[i] - nums[i-1] > 0:
                dp1 = dp0 + 1
            elif nums[i] - nums[i-1] < 0:
                dp0 = dp1 + 1

        res = max(dp0, dp1) + 1
        return res
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值