【SSL】2291分组背包

【SSL】2291分组背包

Time Limit:1000MS

Memory Limit:65536K

Description

有N件物品和一个容量为V的背包。第i件物品的费用是c[i]，价值是w[i]。这些物品被划分为若干组，每组中的物品互相冲突，最多选一件。求解将哪些物品装入背包可使这些物品的费用总和不超过背包容量，且价值总和最大。

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

思路

每组物品有若干种策略：是选择本组的某一件，还是一件都不选。
设f[i][j]表示前i组物品花费空间j能取得的最大权值。
f[i][j]=max(f[i-1][j],f[i-1][j-w[k]]+c[k])(物品k属于组i)。
可以将二维压缩成一维
f[j]=max(f[j],f[j-w[k]]+c[k]);

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<iomanip>
using namespace std;
long long v,n,t,f[1010][1010],head[1010];
struct jgt
{
	long long w,c,p;
};
jgt a[1010];
bool cmp(jgt t1,jgt t2)
{
	return t1.p<t2.p;
}
void input()
{
	long long i,p,w1,c1;
	scanf("%d%d%d",&v,&n,&t);
	for(i=1;i<=n;i++)
		scanf("%d%d%d",&a[i].w,&a[i].c,&a[i].p);
	sort(a+1,a+1+n,cmp);
	p=0;
	for(i=1;i<=n;i++)
	{
		if (a[i].p!=p)
		{
			p=a[i].p;
			head[p]=i;
		}
	}
	return;
}
void DP()
{
	long long i,j,k;
	for(i=1;i<=t;i++)
	{
		for(j=1;j<=v;j++)
		{
			f[i][j]=f[i-1][j];
			if (head[i]!=0)
			{
				for(k=head[i];a[k].p==a[k+1].p;k++)
				{
					if(a[k].w<=j)
						f[i][j]=max(f[i][j],f[i-1][j-a[k].w]+a[k].c);//状态转移方程
				}
				if(a[k].w<=j&&a[k].p!=a[k+1].p)
					f[i][j]=max(f[i][j],f[i-1][j-a[k].w]+a[k].c);
			}
		}
	}
	return;
}
int main()
{
	input();
	DP();
	printf("%d",f[t][v]);
	return 0;
}