LeetCode 416 python 一维数组动态规划

本文介绍了如何使用一维数组动态规划解决将数组分成两个和相等的子集的问题。关键在于判断数组总和是否为偶数以及最大值是否小于等于总和的一半。通过建立状态转移方程,从0开始初始化动态规划数组,逆序遍历数组元素,更新每个状态,最终得出是否存在满足条件的子数组。
摘要由CSDN通过智能技术生成

在这里插入图片描述
一维数组动态规划
题目要求将数组分为两个相等子集的和,那么如果为真,那么这个数组的总和必定是偶数(条件一),且该数组的最大值必不大于该数组和的一半(条件二)。
那么该问题即可转化为求解和为是否有和为total = sum(nums) // 2的子数组问题。
使用一维数组dp = [False] * (total+1),
边界条件:当和为0时,必为真,则dp[0] = True
例:求解和为i (i = (1,2,…total)),
首先遍历每个num in nums,且,为防止重复,i必须逆序遍历。状态转移方程为:当i >= num: dp[i] = dp[i] | dp[i-num]
状态转移方程的意思是:是否有和为i的子数组取决于之前i的状态与i-num的状态。
为何要逆序:
例:[1,2,3,6],dp = [True, False, False, False, False,False, False]
那么遍历完num=1时,若i不逆序,则当i=1时,dp[i] = True(因为dp[i-num] = True),而当i=2时,dp[i] = True,因为dp[i-num] = True(此时i= 2, 因此i-num=1, 而刚更新过的dp[1] = True)。因此i必须逆序,则,较小的和就不会影响较大的和。
为何状态转移方程dp[i] = dp[i] | dp[i-num],
当遍历num in nums时,有可能会出现dp[i-num] = False的情况,然而当前的状态不应该影响以前的状态,若之前遍历的时候dp[i] = True,则若不加 |dp[i],则会出错。

class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        if sum(nums) % 2 == 1:  # 条件一
            return False
        n = sum(nums) // 2
        if max(nums) > n:    # 条件二
            return False
        dp = [False] * (n+1)
        dp[0] = True   # 初始值
        for num in nums:
            for i in range(n, 0, -1):
                if i >= num:
                    dp[i] = dp[i-num] | dp[i]   # 状态转移方程
        return dp[n]

在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值