poj 2392 多重揹包(堆石頭上天)
阿新 • • 發佈:2019-02-14
題意:有很多種石頭,每種石頭的高度是hi,但是此種石頭不能放到ai之上的高度,並且這種石頭有ci個。問用所有這些石頭最高能放多高?
在遍歷高度的時候,可以增加一個flag陣列存放當前種類石頭的使用個數,這樣內層的兩個巢狀迴圈便可以簡化為一個迴圈:
思路:多重揹包。首先對每種石頭能擺到的最高高度進行升序排序。建立一個數組表示下標值所示的高度能否得到。然後遍歷每一種石頭更新陣列。
#include <stdio.h> #include <string.h> #include <stdlib.h> #define N 405 #define M 40000 struct s{ int h,max,num; }s[N]; int used[M+1]; int n; int cmp(const struct s* a,const struct s* b){ return (*a).max-(*b).max; } int main(){ freopen("a.txt","r",stdin); while(scanf("%d",&n)!=EOF){ int i,j,k; memset(used,0,sizeof(used)); for(i = 0;i<n;i++) scanf("%d %d %d",&s[i].h,&s[i].max,&s[i].num); qsort(s,n,sizeof(struct s),cmp); used[0] = 1; for(i = 0;i<n;i++) for(j = s[i].max-s[i].h;j>=0;j--) if(used[j])//如果之前的石頭能夠擺放到j高度 for(k = 1;k<=s[i].num;k++){//從此高度開始擺放第i種石頭,遍歷擺放的個數 if(j+k*s[i].h > s[i].max || used[j+k*s[i].h]) break; used[j+k*s[i].h] = 1; } for(j = M;j>=0;j--) if(used[j]){ printf("%d\n",j); break; } } return 0; }
在遍歷高度的時候,可以增加一個flag陣列存放當前種類石頭的使用個數,這樣內層的兩個巢狀迴圈便可以簡化為一個迴圈:
for(i = 0;i<n;i++){
memset(flag,0,sizeof(flag));
for(j = t[i].h;j<=t[i].maxh;j++)
if(used[j-t[i].h] && !used[j]&&flag[j-t[i].h]<t[i].num){
flag[j] = flag[j-t[i].h]+1;
used[j] = 1;
}
}