0x3f第12天 0-1背包

1.0-1背包问题:

#capacity :背包容量

#w[i]:第i个物体的体积

#v[i]:第i个物体的价值

def dfs(i , c): 定义:选或不选到第i个物体时,目前背包最大价值

i:目前装了i个物体 c:背包剩余容量

回溯公式:

dfs(i,c)=max(dfs(i-1,c),dfs(i-1,c-w[i])+v[i]) 装没装第i个物体的两种情况

边界条件:1.i == 0 没东西可装

  1. 剩余容量不足以装下当前的这个w[ i ]

    #capacity :背包容量

    #w[i]:第i个物体的体积

    #v[i]:第i个物体的价值
    n = len(w)
    def dfs(i,c):
    if i < 0:
    return 0
    if c < w[i]:
    return dfs(i-1,c)
    return max(dfs(i-1,c),dfs(i-1,c-w[i])+v[i])
    dfs(n-1,capacity)




1.1.目标和 问题

1.回溯法+cache

目标和问题可以转换为0-1背包问题,因为最后也是从数组中选出几个数,使其最终和为一个定值

令p:正数和,q:负数和 sums = p-q target = p+q

可以得出 p = (sum+target)//2 即几个数和为定值

dfs(i,t)定义:从i个数中选出和为t的 方案个数

参数 i:从包括i及其往前的数字中选

t:现在要凑的目标和

回溯方程:

dfs(i,t) = dfs(i-1,t) + dfs(i-1,t-nums[i])

边界条件:若sum+target并不是偶数,或者为负数,那就无解,返回 0

边界条件:若当前的nums[i] > t,那就加不进去,return dfs(i-1,t)

初始条件:if i < 0: //没有可选的了,从n已经遍历到开头了

if t==0:return 1 此方案可行

else: return 0 没有恰好为t的方案

代码:

时间复杂度:t*n 空间复杂度::n

时间复杂度你可以把这个dfs函数的执行过程,理解成 "填一张 n 行、t 列的表格":

复制代码
class Solution:
    def findTargetSumWays(self, nums: List[int], target: int) -> int:
        target = sum(nums)+target
        n = len(nums)
        if target%2 != 0 or target<0:
            return 0
        t = target//2
        
        @cache
        def dfs(i,t):
            if i<0:
                if t==0:
                    return 1
                else:
                    return 0
            if nums[i]>t:
                return dfs(i-1,t)
            return dfs(i-1,t) + dfs(i-1,t-nums[i])
        return dfs(n-1,t)

2.递推

先根据回溯公式写出f公式:

dfs(i,t) = dfs(i-1,t) + dfs(i-1,t-nums[i])

f[i][t] = f[i-1][t] + f[i-1][t-nums[i]]数组最好不要出现负数情况

f [ i + 1 ] [ t ] = f [ i ] [ t ] + f [ i ] [ t - nums[ i ] ]

于是自然想到建立一个f[[]]二维数组 为了边界条件,我们的二维数组会多一列、一行

公式:f = [ [0] *(t +1) for _ in range(n+1) ] *的是列数量 for的是行数量

n+1对于前0个,1个...n个元素

t 对应 凑出和为0 ,1 ,...t的数

递归变循环,怎么变,取决于f公式需要什么

f [ i + 1 ] [ t ] = f [ i ] [ t ] + f [ i ] [ t - nums[ i ] ] 可以看出需要 i , t , nums[ i ]

for i,x in enumerate(nums):

for c in range(t+1):

边界条件:nums[i]>t: 在这就是 x>c

时间复杂度:t n 空间复杂度 t n

复制代码
class Solution:
    def findTargetSumWays(self, nums: List[int], target: int) -> int:
        target = sum(nums)+target
        n = len(nums)
        if target%2 != 0 or target<0:
            return 0
        t = target//2

        f = [[0]*(t+1)for _ in range(n+1)]
        f[0][0]=1
        for i,x in enumerate(nums):
            for c in range(t+1):
                if x>c:
                    f[i+1][c] =  f[i][c]
                else:
                    f[i+1][c] = f[i][c]+f[i][c-x]
        return f[n][t]

3.空间优化两个数组

发现f[i+1]只取决于f[i],所以只需要两个数字

之后的数组重新覆盖f0和f1即可

f = [[0]*(t+1)for _ in range(2)] 两行数组

f[i+1]都改成f[(i+1)%2]

f[i]都改成f[i%2]

空间复杂度为 t

4.空间优化一个数组

和i有关的都删掉,遍历target的时候倒着遍历,从t遍历至x,倒序到 x,只处理 c≥x 的场景,避免重复选,最开始的边界条件

时间复杂度 tn 空间复杂度t

复制代码
class Solution:
    def findTargetSumWays(self, nums: List[int], target: int) -> int:
        target = sum(nums)+target
        n = len(nums)
        if target%2 != 0 or target<0:
            return 0
        t = target//2

        f =[0]*(t+1)
        f[0]=1
        for x in nums:
            for c in range(t,x-1,-1):
                f[c] = f[c]+f[c-x]
        return f[t]



2.完全背包

每个物品可以重复选,回溯公式修改:

dfs(i,c)=max(dfs(i-1,c),dfs(i-1, c-w[i])+v[i])

dfs (i,c) = min(dfs(i-1,c), dfs(i, c-w[i])+v[i])选了,但还可以选,所以i不变

2.1 零钱兑换

1.回溯法+cache

dfs(i,c)定义:用前i种硬币,凑c块钱,最少用几个硬币

i:前i种硬币

c:还需要凑c块钱

回溯方程:

dfs(i,c) = min(dfs(i-1,c) + dfs(i-,c-coins[i])+1)

边界条件:

1.若 coins[i]>c,return dfs(i-1,c) 这类硬币滚蛋

2.若i < 0 and c!=0,说明还没凑完,return inf 无解

起始条件:

若 i<0 and c==0,return 0 所有硬币都处理完了,且剩余要凑的金额是 0不需要再用任何硬币

i < 0c == 0 → 返回 0
  • 含义:所有硬币都处理完了,且剩余要凑的金额是 0(已经凑够了);
  • 为什么返回 0?因为 "没有硬币可选" 的情况下,凑够了金额 0,不需要再用任何硬币 → 最少硬币数就是 0;
  • 举例子:比如你要凑金额 0,不管有没有硬币,都只需要 0 个硬币(啥都不用选),这是所有最值问题的 "基础锚点"。
复制代码
  class Solution:
      def coinChange(self, coins: List[int], amount: int) -> int:
          n = len(coins)
          @cache
          def dfs(i,c):
              if i<0:
                  if c==0:return 0
                  else:   return inf
              if coins[i] > c:
                  return dfs(i-1,c)
              return min(dfs(i-1,c),dfs(i,c-coins[i])+1)
          ans = dfs(n-1,amount)
          if ans < inf:
              return ans
          else: 
              return -1
相关推荐
艾醒2 小时前
大模型原理剖析——突破LLM效率瓶颈:多标记预测(MTP)技术深度解析与实战
算法
智驱力人工智能2 小时前
森林防火无人机火焰监测系统 构建“天空地”一体化智能防火体系 无人机火焰检测,支持红色火焰检测 城市高层建筑无人机火焰识别
人工智能·深度学习·opencv·算法·目标检测·无人机·边缘计算
Bruce_kaizy2 小时前
c++图论————最短路之Floyd&Dijkstra算法
c++·算法·图论
电商API&Tina2 小时前
跨境电商速卖通(AliExpress)数据采集与 API 接口接入全方案
大数据·开发语言·前端·数据库·人工智能·python
WBluuue2 小时前
AtCoder Beginner Contest 437(ABCDEF)
c++·算法
郝学胜-神的一滴2 小时前
Linux 下循环创建多线程:深入解析与实践指南
linux·服务器·c++·程序人生·算法·设计模式
工业互联网专业2 小时前
基于python的旅游景点推荐系统_flask+spider
python·flask·毕业设计·源码·课程设计·spider·旅游景点推荐系统
编程乐学(Arfan开发工程师)2 小时前
信息收集与分析详解:渗透测试的侦察兵 (CISP-PTE 核心技能)
java·开发语言·javascript·python
superman超哥2 小时前
仓颉语言中异常处理入门的深度剖析与工程实践
c语言·开发语言·c++·python·仓颉