题意:一个人买商品,必须每个种类的商品至少买一件,同时保证买到的价值尽可能大,如果不能满意,输出Impossible。
分组背包的变种,普通分组背包是一个组里面最大取一件,这题是至少取一件。
设计状态dp[i][j]代表前i组容量为j的最大价值。由于一组里面有多个物品,所以状态转移可以是前一组少取一个,即dp[i-1][p-g[i][j].v]+g[i][j].w,也可以是当前组之前去过的少取一种,即dp[i][p-g[i][j].v]+g[i][j].w。
因为每种必须取一个,所以要注意初始化和判断。
AC代码如下:
#include <iostream>
#include <vector>
#include <cstdio>
#include <iterator>
using namespace std;
typedef struct{
int label, value;
}Node;
inline int max( int a, int b ){
return ( a > b ? a : b );
}
int main(){
int N, M, K;
int a, b, c;
vector<Node> sneaker[11];
int dp[11][10001];
while( scanf( "%d%d%d", &N, &M, &K ) != EOF ){
for( int i = 0; i <= 10; i++ ){
while( !sneaker[i].empty() ){
sneaker[i].pop_back();
}
}
for( int i = 0; i < N; i++ ){
cin >> a >> b >> c;
Node temp;
temp.label = b;
temp.value = c;
sneaker[a].push_back( temp );
}
memset( dp, -1, sizeof( dp ) );
memset( dp[0], 0, sizeof( dp[0] ) );
for( int i = 1; i <= K; i++ ){
for( vector<Node>::iterator j = sneaker[i].begin(); j != sneaker[i].end(); j++ ){
for( int k = M; k >= ( j->label ); k-- ){
if( dp[i][k-j->label] != -1 ){
dp[i][k] = max( dp[i][k], dp[i][k-j->label] + j->value );
}
if( dp[i-1][k-j->label] != -1 ){
dp[i][k] = max( dp[i][k], dp[i-1][k-j->label] + j->value );
}
}
}
}
if( dp[K][M] == -1 ){
puts( "Impossible" );
}else{
cout << dp[K][M] << endl;
}
}
return 0;
}