1. 程式人生 > >POJ 2392 Space Elevator(帶限制條件的多重揹包)

POJ 2392 Space Elevator(帶限制條件的多重揹包)

大意:一群牛用石塊堆天梯,不同的石塊有不同的高度和最高的堆疊高度,求最終的高度。

分析:覺得是多重揹包,但是有了高度的限制。看了別人寫的程式碼半天才緩過來。啊,這樣處理。DP路漫漫。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=4e4+10;
struct node{
    int h,hmax,c;
}stone[410];
int cmp(node a,node b){
     return a.hmax<b.hmax;
}
int dp[N],sum[N];
int main()
{
    int n;
    while(cin>>n){
        for(int i=0;i<n;i++){
            scanf("%d%d%d",&stone[i].h,&stone[i].hmax,&stone[i].c);
        }
        sort(stone,stone+n,cmp);
        memset(dp,0,sizeof(dp));
        dp[0]=1;
        int ans=0;
        for(int i=0;i<n;i++){
            memset(sum,0,sizeof(sum));
            for(int j=stone[i].h;j<=stone[i].hmax;j++){
                if(dp[j]==0&&dp[j-stone[i].h]&&sum[j-stone[i].h]<stone[i].c){
                    dp[j]=1;
                    sum[j]=sum[j-stone[i].h]+1;
                    ans=ans>j?ans:j;
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}