poj 2392 Space Elevator (多重背包)

http://poj.org/problem?id=2392 背包没有容量，对于每种木块，其能达到的最大高度a便可做为其容量。先用二进制优化减少数量，然后根据a的值排序，01背包时可以以每种木块可达到的最大高度为最大容量。RE一次，因为val的大小只开了401...这里是个易错点啊，二进制优化后的数量不确定。

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#define Max(a, b) a>b?a:b
int dp[40005] ;
struct Block{
int h ;
int a ;
}val[4001] ;
int cmp(const void *a, const void *b){
return (*(Block *)a).a - (*(Block *)b).a ;
}
int main(){
int n, i, j, count, h, c, a, max ;
while(~scanf("%d", &n)){
count = 0 ;
memset(dp, 0, sizeof(dp)) ;
while(n--){
scanf("%d%d%d", &h, &a, &c) ;
//二进制优化
i = 1 ;
while(c>=i){
val[count].h = h * i ;
val[count++].a = a ;
c -= i ;
i *= 2 ;
}
if(c){
val[count].h = h * c ;
val[count++].a = a ;
}
}
qsort(val, count, sizeof(val[0]), cmp) ;
//01背包求解
max = 0 ;
for(i=0; i<count; i++)
for(j=val[i].a; j>=val[i].h; j--)
dp[j] = Max(dp[j], dp[j-val[i].h]+val[i].h) ;
for(j=0; j<=val[i-1].a; j++)
if(dp[j]>max) max = dp[j] ;
printf("%d\n", max) ;
}
return 0 ;｝