1. 程式人生 > >poj 2392 多重揹包(堆石頭上天)

poj 2392 多重揹包(堆石頭上天)

題意:有很多種石頭,每種石頭的高度是hi,但是此種石頭不能放到ai之上的高度,並且這種石頭有ci個。問用所有這些石頭最高能放多高?

思路:多重揹包。首先對每種石頭能擺到的最高高度進行升序排序。建立一個數組表示下標值所示的高度能否得到。然後遍歷每一種石頭更新陣列。

#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;
		}
}