分组背包

Posted 2020-09-13 小时のblog

Problem Description

一个旅行者有一个最多能用V公斤的背包，现在有n件物品，它们的重量分别是W1，W2，...,Wn，它们的价值分别为C1,C2,...,Cn。这些物品被划分为若干组，每组中的物品互相冲突，最多选一件。求解将哪些物品装入背包可使这些物品的费用总和不超过背包容量，且价值总和最大。

Input

输入有多组数据，每组数据的第一行：三个整数，V(背包容量，V<=200)，N(物品数量，N<=30)和T(最大组号，T<=10)； 第2..N+1行：每行三个整数Wi,Ci,P，表示每个物品的重量，价值，所属组号。

Output

对于每组数据输出仅一行，一个数，表示最大总价值。

Sample Input

10 6 3
2 1 1
3 3 1
4 8 2
6 9 2
2 8 3
3 9 3

Sample Output

20

【代码】

#include<iostream>
#include<cstdio>
using namespace std;
int w[500],v[500],a[500][500],f[400];
int  main()
{
    int vv,n,t;
    scanf("%d%d%d",&vv,&n,&t);
    for(int i=1;i<=n;i++)
    {
        int p;
        scanf("%d%d%d",&w[i],&v[i],&p);
        a[p][++a[p][0]]=i;
    }
    for(int i=1;i<=t;i++)
    {
        for(int j=vv;j>=0;j--)
        {
            for(int k=1;k<=a[i][0];k++)
            {
                if(w[a[i][k]]<=j)
                f[j]=max(f[j],f[j-w[a[i][k]]]+v[a[i][k]]);
            }
        }
    }
    printf("%d",f[vv]);
    return 0;
}