分组背包问题

分组背包问题

 

 

 

 二维：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 110;

int dp[maxn][maxn];//dp[i][j]:只从前i组物品中选, 当前体积小于等于j的最大值
int v[maxn][maxn], w[maxn][maxn], s[maxn]; //v为体积, w为价值, s代表第i组物品的个数
int n,m,k;

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
        for(int j=0;j<s[i];j++){
            scanf("%d%d",&v[i][j], &w[i][j]);
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<=m;j++){
            dp[i][j] = dp[i-1][j];//第i组不选
            for(int k=0;k<s[i];k++){//第i组选哪个
                if( j>=v[i][k] ){
                    dp[i][j] = max(dp[i][j], dp[i-1][j-v[i][k]]+w[i][k]);
                }
            }
        }
    }
    printf("%d\n",dp[n][m]);
    return 0;
}

一维：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 110;

int dp[maxn];//仿照01背包滚动数组
int v[maxn][maxn], w[maxn][maxn], s[maxn]; //v为体积, w为价值, s代表第i组物品的个数
int n,m,k;

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
        for(int j=0;j<s[i];j++){
            scanf("%d%d",&v[i][j], &w[i][j]);
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=m;j>=0;j--){
            for(int k=0;k<s[i];k++){//第i组选哪个
                if( j>=v[i][k] ){
                    dp[j] = max(dp[j], dp[j-v[i][k]]+w[i][k]);
                }
            }
        }
    }
    printf("%d\n",dp[m]);
    return 0;
}