代码随想录算法训练营第四十七天(动态规划篇)| 416. 分割等和子集

本文主要是介绍代码随想录算法训练营第四十七天(动态规划篇)| 416. 分割等和子集,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

416. 分割等和子集

题目链接:416. 分割等和子集 - 力扣(LeetCode)

思路

回溯——超时

首先想到之前的回溯算法,寻找数组中加和等于sum(nums)/2的子集,但对于大数组超时了:

class Solution(object):def backtracking(self, nums, startIdx, curSum):if curSum > sum(nums)/2:return if curSum == sum(nums)/2 and sum(nums)%2 == 0:return Truefor i in range(startIdx, len(nums)):curSum += nums[i]if self.backtracking(nums, i+1, curSum) == True:return TruecurSum -= nums[i]return Falsedef canPartition(self, nums):return self.backtracking(nums, 0, 0)

动态规划

可以把数组中的每个元素看作一个物体,那么寻找符合要求的子集相当于把将所选的物体装入一定容量的背包。根据之前的学习,我们知道如何求定容背包能装的最大价值,如果把每一个元素的数值同时当成该物体的重量和价值,就可以用01背包求解。

如果容量为sum/2的背包能装的最大价值为sum/2(即可以被装满),则说明这个数组可以被分割成两个子集。

e,g, 对于数组[1, 2, 5], 有三个物体,它们的重量和价值分别为1,2,5,那对于容量为sum/2 = 4的背包,最多只能装1+2 = 3的价值,说明不能被分割。

1. dp数组定义

容量为j的背包,所背的物品价值最大可以为dp[j]。

若dp[j] == j 说明集合中的某子集总和正好可以凑成总和j;如果dp[sum/2] = sum/2,说明数组可以被分割成等和子集。

2. dp递推公式

01背包的递推公式为:dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);

本题,相当于背包里放入物品i的重量是nums[i],其价值也是nums[i]。

所以递推公式:dp[j] = max(dp[j], dp[j - nums[i]] + nums[i]);

3. 初始条件

当背包的容量为0时,能装的最大价值为0,所以dp[0]=0。如果题目给的价值都是正整数那么非0下标都初始化为0就可以了,如果题目给的价值有负数,那么非0下标就要初始化为负无穷。

4. 遍历顺序

根据01背包的滚动数组的方法:

代码随想录算法训练营第四十六天(动态规划篇)|01背包(滚动数组方法)-CSDN博客

如果使用一维dp数组,物品遍历的for循环放在外层,遍历背包容量的for循环放在内层,且内层for循环倒序遍历!

5. 举例推导dp数组

代码实现

class Solution(object):def canPartition(self, nums): # 剪枝:如果数组总和为奇数,则不能分割成等和子集if sum(nums)%2 == 1:return Falsetarget = sum(nums)/2dp = [0] * (target + 1)for num in nums:for j in range(target, num-1, -1): # 等于省略了之前if的条件之前的题需要trace weight[i]的值dp[j] = max(dp[j], dp[j-num] + num)return dp[target] == target

需要注意,在循环背包容量时,我们用

for j in range(target, num-1, -1):

因为对于重量(价值)为num的物体只能放在容量大于他自身重量的包里。这一个条件在之前的01背包代码是由下面的if语句实现:

if weight[i] > j:dp[j] = dp[j]

如果当前重量大于背包容量,就选择不装物品i,这相当于不考虑背包容量小于当前物体的值,即不在for循环中遍历。事实上之前的代码:

for i in range(objNum): # 遍历物体for j in range(bagWeight, 0, -1):  #遍历背包容量if weight[i] > j:dp[j] = dp[j]else:dp[j] = max(dp[j], dp[j - weight[i]] + value[i])

也可以改为:

for i in range(objNum): # 遍历物体for j in range(bagWeight, weight[i] - 1, -1):  # 遍历背包容量dp[j] = max(dp[j], dp[j - weight[i]] + value[i])

这篇关于代码随想录算法训练营第四十七天(动态规划篇)| 416. 分割等和子集的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/695374

相关文章

利用Python调试串口的示例代码

《利用Python调试串口的示例代码》在嵌入式开发、物联网设备调试过程中,串口通信是最基础的调试手段本文将带你用Python+ttkbootstrap打造一款高颜值、多功能的串口调试助手,需要的可以了... 目录概述:为什么需要专业的串口调试工具项目架构设计1.1 技术栈选型1.2 关键类说明1.3 线程模

SpringBoot基于配置实现短信服务策略的动态切换

《SpringBoot基于配置实现短信服务策略的动态切换》这篇文章主要为大家详细介绍了SpringBoot在接入多个短信服务商(如阿里云、腾讯云、华为云)后,如何根据配置或环境切换使用不同的服务商,需... 目录目标功能示例配置(application.yml)配置类绑定短信发送策略接口示例:阿里云 & 腾

Python Transformers库(NLP处理库)案例代码讲解

《PythonTransformers库(NLP处理库)案例代码讲解》本文介绍transformers库的全面讲解,包含基础知识、高级用法、案例代码及学习路径,内容经过组织,适合不同阶段的学习者,对... 目录一、基础知识1. Transformers 库简介2. 安装与环境配置3. 快速上手示例二、核心模

Java的栈与队列实现代码解析

《Java的栈与队列实现代码解析》栈是常见的线性数据结构,栈的特点是以先进后出的形式,后进先出,先进后出,分为栈底和栈顶,栈应用于内存的分配,表达式求值,存储临时的数据和方法的调用等,本文给大家介绍J... 目录栈的概念(Stack)栈的实现代码队列(Queue)模拟实现队列(双链表实现)循环队列(循环数组

Python实现图片分割的多种方法总结

《Python实现图片分割的多种方法总结》图片分割是图像处理中的一个重要任务,它的目标是将图像划分为多个区域或者对象,本文为大家整理了一些常用的分割方法,大家可以根据需求自行选择... 目录1. 基于传统图像处理的分割方法(1) 使用固定阈值分割图片(2) 自适应阈值分割(3) 使用图像边缘检测分割(4)

使用Java将DOCX文档解析为Markdown文档的代码实现

《使用Java将DOCX文档解析为Markdown文档的代码实现》在现代文档处理中,Markdown(MD)因其简洁的语法和良好的可读性,逐渐成为开发者、技术写作者和内容创作者的首选格式,然而,许多文... 目录引言1. 工具和库介绍2. 安装依赖库3. 使用Apache POI解析DOCX文档4. 将解析

C++使用printf语句实现进制转换的示例代码

《C++使用printf语句实现进制转换的示例代码》在C语言中,printf函数可以直接实现部分进制转换功能,通过格式说明符(formatspecifier)快速输出不同进制的数值,下面给大家分享C+... 目录一、printf 原生支持的进制转换1. 十进制、八进制、十六进制转换2. 显示进制前缀3. 指

Python如何将大TXT文件分割成4KB小文件

《Python如何将大TXT文件分割成4KB小文件》处理大文本文件是程序员经常遇到的挑战,特别是当我们需要把一个几百MB甚至几个GB的TXT文件分割成小块时,下面我们来聊聊如何用Python自动完成这... 目录为什么需要分割TXT文件基础版:按行分割进阶版:精确控制文件大小完美解决方案:支持UTF-8编码

openCV中KNN算法的实现

《openCV中KNN算法的实现》KNN算法是一种简单且常用的分类算法,本文主要介绍了openCV中KNN算法的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的... 目录KNN算法流程使用OpenCV实现KNNOpenCV 是一个开源的跨平台计算机视觉库,它提供了各

使用Python实现全能手机虚拟键盘的示例代码

《使用Python实现全能手机虚拟键盘的示例代码》在数字化办公时代,你是否遇到过这样的场景:会议室投影电脑突然键盘失灵、躺在沙发上想远程控制书房电脑、或者需要给长辈远程协助操作?今天我要分享的Pyth... 目录一、项目概述:不止于键盘的远程控制方案1.1 创新价值1.2 技术栈全景二、需求实现步骤一、需求