题意:n个物品,k个组,每个物品只能属于一个组,求至少每个组选一个物品实现的最大价值
这是我的第一道分组背包,这个和背包九讲里面的那个典型例子不同,想了好久,都感觉没有办法确定每个组至少选一个
所以看了题解,我用dp[i][j] 表示前i组物品,放在容量为j的背包里面实现的最大价值,这里我感觉到了要求恰好装满i组物品,求最大值的二维费用背包,
然后就写出了代码,状态转移方程dp[i][j] = max(dp[i][j], dp[i][j - s[i][k].cost] + s[i][k].value, dp[i][j - s[i-1][k].cost] + s[i][k].value)//其中s[i][j] 表示第i组的第j个物品,也就是
存的是输入信息。
这里还要啰嗦一下,初始化问题,以前我们做过,恰好装满求最大值,初始化dp[0] = 0,dp[1------]初始化为-OO,所以这里dp[0][i] = 0 ,dp[1----][i]初始化为-OO
代码如下:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<stdio.h>
#include<stdlib.h>
#include<ctype.h>
#include<time.h>
#include<math.h>
#define inf 0x7fffffff
#define eps 1e-9
#define pi acos(-1.0)
#define P system("pause")
using namespace std;
struct node
{
int cost,value;
};
vector<node> s[15];
int dp[15][10005];//dp[i][j]表示前i组物品放在容量为j中的背包中实现的最大价值
int main()
{
//freopen("input.txt","r",stdin);
//freopen("output.txt","w",stdout);
int n,m,k;
while(scanf("%d%d%d",&n,&m,&k) != EOF)
{
int i,j;
for(i = 0; i < 12; i++)
s[i].clear();
for(i = 0 ; i < n; i++)
{
int a, b, c;
node temp;
scanf("%d%d%d",&a,&b,&c);
temp.cost = b; temp.value = c;
s[a].push_back(temp);
}
for(i = 0; i <= m; i++)
dp[0][i] = 0;
for(i = 1; i <= k; i++)
for(j = 0; j <= m; j++)
dp[i][j] = -inf;
for(i = 1; i <= k; i++)//第i组物品
{
int len = s[i].size();
for(j = 0; j < len; j++)
for(int z = m; z >= s[i][j].cost; z--)//这里是0-1背包
{
dp[i][z] = max(dp[i][z], dp[i][z-s[i][j].cost]+s[i][j].value);
dp[i][z] = max(dp[i][z], dp[i-1][z-s[i][j].cost]+s[i][j].value);
}
}
if(dp[k][m] < 0) printf("Impossible\n");
else printf("%d\n",dp[k][m]);
}
/*这题让我有了将分组背包转为二维背包的感觉*/
return 0;
}