数学建模社区-数学中国

标题: python代码解决0-1背包问题 [打印本页]

作者: 2744557306    时间: 2023-11-7 11:17
标题: python代码解决0-1背包问题
一、首先介绍一下0-1背包问题:
! ^& l# K% ^+ n9 O$ A* O- o3 g
        0-1背包问题(0-1 Knapsack Problem)是一个经典的组合优化问题,通常在计算机科学和运筹学中讨论。这个问题涉及到一个背包和一组物品,每个物品都有一个特定的重量和价值。问题的目标是在给定背包的最大容量下,选择一组物品放入背包,以使得所选物品的总重量不超过背包的容量,同时最大化这些物品的总价值。
' s2 O, ]7 K& D% q# R4 r6 s         0-1背包问题的名称中的“0-1”表示每个物品要么完全放入背包(选择)要么完全不放入背包(不选择),不能部分放入。这是问题的一个关键特征,与分数背包问题不同,分数背包问题允许部分放入物品。: p  O( }5 f* \1 c8 M$ T
问题的形式描述如下:
$ Y. B# \, I0 M
                 1.给定一个固定容量的背包,通常表示为一个正整数W(背包的最大承载重量)。
" O4 W( V- |. t4 Z                 2.给定一组物品,每个物品都有两个属性:重量(weight)和价值(value)。
$ }, ~7 G0 Z. J: n4 N) o                 3.对每个物品,你可以选择将其放入背包(选择)或不放入背包(不选择)。
9 ?, @# l7 F$ l- e. r                 4.每个物品只能选择一次,即要么放入背包,要么不放入。' E4 o6 Z" W9 {
                 5.目标是选择一个物品组合,使得它们的总重量不超过背包容量W,同时使它们的总价值最大化。4 {) ~  E5 v/ Y  x* M
1 k) `# K- X* t1 \+ f% Y
解决0-1背包问题的一种常见方法是使用动态规划(Dynamic Programming)算法。这个问题有广泛的应用,包括资源分配、排程问题、投资组合优化等领域。它还是计算复杂性理论中的一个经典问题,通常被用来说明NP难问题的概念。4 b8 p5 }. N& ^. E

% g3 j3 Z6 e9 n, C$ }' P
$ v& q5 ~, ]- e3 k% j8 }二、 介绍代码
: r+ M  G2 l+ U3 Q8 m这段代码是一个Python实现的0-1背包问题的解决方法,使用了动态规划算法来找到最优解。以下是对代码的详细解释:
  1. def knapsack(v, w, n, capacity):
    ' w% z& t% D: f* j
  2.     i = 0
    ) ]  M+ D9 _3 C& r2 M9 [
  3.     capacity = capacity + 1  # 初始化背包容量最大值# s, E  ?2 l3 z) V
  4.     m = np.zeros((n, capacity))  # 初始化9 s4 T5 v7 Z4 A/ r2 f' A; z
  5.     x = np.zeros(n)
复制代码
1.v 是一个长度为 n 的列表,表示每个物品的价值。4 U/ b9 |2 L7 b% j$ W
2.w 是一个长度为 n 的列表,表示每个物品的重量。4 a' }' M6 }% b. ~- h
3.n 表示物品的数量。
; D2 e& N* f1 T4.capacity 表示背包的容量。
  n9 p; }$ u  r' c2 U$ M6 ~6 P$ n; i' M0 W! _& D( ?! p# ~7 x2 b  l. N
代码首先初始化了一个二维数组 m 作为动态规划表,其中 m[j] 表示在考虑前 i 个物品时,背包容量为 j 时可以获得的最大总价值。数组 x 用来存储最终的解,x 表示是否选择第 i 个物品。
  1.     for i in range(n):
    - J8 ]$ v% T! U9 [4 F
  2.         for j in range(capacity):! a$ Z/ Z" }6 l' j- R  T; o' v
  3.             if (j >= w[i]):" @4 x6 \) Y9 i* q8 _
  4.                 m[i][j] = max(m[i - 1][j], m[i - 1][j - w[i]] + v[i]); e2 M9 t! y4 {: ?: ~7 ^- w# [7 @$ @
  5.             else:
    * T/ Y  c' C# T; z/ K
  6.                 m[i][j] = m[i - 1][j]
复制代码
在这个部分,代码使用了一个嵌套的循环,遍历了所有物品和不同的背包容量。对于每个物品 i 和容量 j,代码计算了两种情况下的最大总价值:: n# |9 |& l' I: _  \  y, Z
5.如果当前物品的重量 w 小于等于当前容量 j,那么可以选择将第 i 个物品放入背包,此时总价值为 m[i-1][j-w] + v,或者选择不放入,此时总价值为 m[i-1][j]。代码选择其中较大的值作为 m[j]。
8 d' \3 H7 [- U" O6.如果当前物品的重量 w 大于当前容量 j,则无法放入物品,所以总价值等于上一行的值 m[i-1][j]。6 G0 c$ q9 G, p( y) I

( y3 Y. h) V) u, S* ]这个循环填充了动态规划表 m,最终 m[n-1][capacity] 包含了问题的最优解,即在给定容量下可以获得的最大总价值。
  1.     capacity = capacity - 1
    # p/ i6 S6 _1 {2 r, k
  2.     for i in range(n - 1, 0, -1):4 ?2 b: f0 V1 a1 X2 h
  3.         if (m[i][capacity] == m[i - 1][capacity]):: p3 a1 _1 Y% T9 o
  4.             x[i] = 05 ~: ~8 e- q: R! J% ]
  5.         else:
    % U  o& e3 o5 U, c
  6.             x[i] = 1
    ! v# _/ z- r- P# B3 q" f0 R
  7.             capacity -= w[i]( Y0 \  F+ I2 e5 ^$ s  q( d$ F
  8.     x[0] = 1 if (m[1][capacity] > 0) else 0
复制代码
在这一部分,代码反向遍历动态规划表,从最后一行向前找到解的路径。如果 m[capacity] 等于 m[i-1][capacity],表示第 i 个物品没有放入背包,否则放入背包,并更新剩余容量 capacity。
  1.     weight = 0
    " F( z' l: v  A; Y( n
  2.     value = 0
    " X! d: P7 P: H9 k" u. Q2 K, l1 F; A
  3.     print('装载的物品编号为:')
    # R( c) P# \! I4 g. X% ?
  4.     for i in range(len(x)):
    & Z9 }4 N' j4 Q1 V, n
  5.         if (x[i] == 1):5 C$ i% x$ }1 I& ~  t% X
  6.             weight = weight + w[i]9 h4 ?; Z0 L* o8 I4 {
  7.             value = value + v[i]( M+ I% J  w7 M1 \$ x- e1 z
  8.             print(' ', i + 1)" ~+ X( Q; F& W
  9.     print('装载的物品重量为:')( v( `/ F* B: r8 w, {
  10.     print(weight)/ S; _5 X7 j( ]# e
  11.     print('装入的物品价值为:')' H7 f/ X+ ]* l8 E8 t, q9 ~
  12.     print(value)
    6 |2 H1 x, Z/ w, X
  13.     return m
复制代码
最后,代码计算了被选择的物品的总重量和总价值,并将它们打印出来。函数返回动态规划表 m。) M2 f* c0 i* [3 n9 G( t8 ]& o9 P
这段代码实现了0-1背包问题的解决方法,它通过动态规划算法找到最优解,即在给定背包容量下可以获得的最大总价值,以及选择哪些物品放入背包。3 h' q) t% V! g7 ^. U: _& Q
, u# G$ e( @+ M, `) {
最后在函数的输入文件中,如下例,第一行物品数量为:5 背包载重量为:10物品的重量列表为:[2, 2, 6, 5, 4] 物品的价值列表为: [6, 3, 5, 4, 6]! n8 T7 N; m' g: K: @' Z4 o  ]
VeryCapture_20231107110045.jpg

- R2 D* j: S' q+ W/ m7 |; Q% o
# l- m2 w& o" x8 f8 i8 {' O
接下来展示我们的输出结果:

7 F& t6 c, @$ w
VeryCapture_20231107110508.jpg

6 [# k' c) V4 v. G* b; D
1 c$ Q0 m) w7 K5 Y

: u) Y) \1 \) `4 T
具体代码如下:
  ~4 M5 ]! [% y. \1 o4 C
) r& v; K9 u5 V" t* n

7 L( q: F& g; i3 g$ ?$ @5 m2 U( E5 M3 f! b# F- o

% C7 I) R3 o$ N3 i7 t6 m( E; `, k8 @7 e+ `. L0 ]/ B

基础0-1背包问题(动态规划).rar

3.58 KB, 下载次数: 0, 下载积分: 体力 -2 点

售价: 2 点体力  [记录]  [购买]






欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) Powered by Discuz! X2.5