编程训练——完全背包问题

题目描述

$n$个物品，重量分别为$w_i$和$v_i$，从这些物品中挑选总重量不超过$W$的物品，求出挑选物品价值综合的最大值，每种物品都可以挑选任意多件。（01背包问题是每种物品只能挑选一件）

文中将以数组$w$存储重量，数组$v$存储价值，$w[i]$和$v[i]$分别代表第i号物品（或者叫第$i+1$个物品）的重量和价值。

正整数$n$不超过100，每个物品的重量和价值都属于$[1,100]$，总重量为$[1,10000]$。

样例输入

3
3 4 2
4 5 3
7

总共3个物品，分别重量为3、4、2，分别价值为4、5、3，求挑选出不超过重量7的物品的最大总价值。

样例输出

10

选1个0号，2个2号。

算法

使用动态规划，$dp[i+1][j]$代表从前$i+1$个物品中挑出重量不超过$j$的物品的最大总价值，前$i+1$个物品指的是$0～i$号物品。

对于$dp[i+1][j]$，有两种选择，一种是不挑选$i$号物品（或者叫第$i+1$个物品），一种是至少挑选一个第$i$号物品。如果不挑选第i号物品，那么$dp[i+1][j]=dp[i][j]$；如果挑选至少一个第i号物品，那么$dp[i+1][j]=dp[i+1][j-w[i]]+v[i]$。

综上所述，
d p [ i ] [ j ] = { d p [ i + 1 ] [ j ] = m a x { d p [ i ] [ j ] , d p [ i + 1 ] [ j − w [ i ] ] + v [ i ] } , j ≥ w [ i ] d p [ i + 1 ] [ j ] = d p [ i ] [ j ] , j < w [ i ] dp[i][j]=\left\{ \begin{aligned} dp[i+1][j]=max\{dp[i][j],dp[i+1][j-w[i]]+v[i]\},j≥w[i]\\ dp[i+1][j]=dp[i][j],j<w[i] \end{aligned} \right. dp[i][j]={dp[i+1][j]=max{dp[i][j],dp[i+1][j−w[i]]+v[i]},j≥w[i]dp[i+1][j]=dp[i][j],j<w[i]​
其中注意判断是否有$j\ge w[i]$，初始值为$dp[0][j]=0$（前0个物品，即没有物品，自然就没有价值）。

代码

#include<stdio.h>

#define maxn 105
#define maxW 10005

int max(int a, int b){
    return (a>b)?a:b;
}

int main(){
    int w[maxn], v[maxn];   // 分别存储重量和价值
    int n, W;   // 物品数量和最大总重量
    int dp[maxn][maxW];     // 严谨来说是dp[maxn+1][maxW]，但因习惯将#define定义的数字定大一点，所以这样写也没关系。
    while(scanf("%d", &n)==1){
        for(int i = 0;i < n;i++){
            scanf("%d", &w[i]);
        }
        for(int i = 0;i < n;i++){
            scanf("%d", &v[i]);
        }
        scanf("%d", &W);
        for(int j = 0;j < W;j++){
            dp[0][j] = 0;
        }
        for(int i = 1;i <= n;i++){
            for(int j = 0;j <= W;j++){
                if(j>=w[i-1]){
                    dp[i][j] = max(dp[i-1][j],dp[i][j-w[i-1]]+v[i-1]);
                }
                else dp[i][j] = dp[i-1][j];
            }
        }
        printf("%d\n", dp[n][W]);
    }

    return 0;
}