1. 程式人生 > >[完全揹包dp]Space Elevator POJ

[完全揹包dp]Space Elevator POJ

poj.org/problem?id=2392

有一群奶牛想到太空去,他們有k中型別的石頭,每一類石頭高h,石頭能達到的高度c,以及它的數量a,在做揹包前需要對石塊能到達的最大高度(a)進行排序,而且每種磚塊都有一個限制條件,就是說以該種磚塊結束的最大高度H不能超過某個高度,不同磚塊的高度不同。求最高的高度是多少。

自己以前寫 就是暴力把每個高度用for k->c 塞進數組裡面 然後徹底當01揹包寫 還有書上在dp裡層多寫個for k->c 一樣的道理

記得以前在poj上遇到會超時的題 當時就見到用used的了 不過寫這題的時候並沒有記住orz

現在想想多個 used陣列 同樣都是轉01揹包了 不過更快了 現在算是徹底記下了 

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <list> 
using namespace std;
typedef long long ll;

const int maxn = 400+5;
const int maxv = 400*100+5;
const int mod = 1e9+7;
const int INF = 0x3f3f3f3f;
const double PI=acos(-1.0);

bool dp[maxv];
int used[maxv]; // 比起再寫一個 for k->c 顯然用這個節約了時間 

struct node{
	int h,a,c;
	bool operator < (const node&n)const {
        return a<n.a;
    }
}a[maxn];

int main(){
	std::ios::sync_with_stdio(false);
	int n,maxh=0;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i].h>>a[i].a>>a[i].c;
		maxh=max(maxh,a[i].a);
	}
	sort(a,a+n);// 按 高度限制有小到大排序 
	memset(dp,0,sizeof(dp));
	dp[0]=1;
	for(int i=0;i<n;i++){
		memset(used,0,sizeof(used));// 此後多重揹包問題求解  
		for(int j=a[i].h;j<=a[i].a;j++){
			if(!dp[j]&&dp[j-a[i].h]&&used[j-a[i].h]<a[i].c){
				used[j]=used[j-a[i].h]+1;
				dp[j]=1;
			}
		}
	}
	for(int i=maxh;i>=0;i--)
		if(dp[i]){
			cout<<i<<endl;
			break;
		}
	return 0;
} 

越做dp越發覺 自己的IQ不足 orz