Algoogle

Algorithm for Programming Contest

PKU 2392 Space Elevator

Category: PKU Tag: dp

Space Elevator

問題概要


解法


aの大きさ順にソートして, 小さいやつからブロックを置くのは明らかにわかる.
あるブロックまでで作れるのはその前のブロックが作れるものにhを足したもの
またそれをc回繰り返したもの.
それをDPすればいい.
dp[i] := 高さ i が作れる

コード


(2392.cpp) download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#include <algorithm>
#include <cstdio>

using namespace std;

struct S{
    int h, a, c;
    S(){}
    S(int h, int a, int c):h(h),a(a),c(c){}
    bool operator<(const S &s) const{ return a < s.a;}
};

int K;
int h, a, c;
S block[512];
bool dp[40000];

int main(){
    scanf("%d", &K);
    for(int i = 0; i < K; i++){
        scanf("%d%d%d", &h, &a, &c);
        block[i] = S(h, a, c);
    }
    sort(block, block+K);

    dp[0] = 1;
    for(int i = 0; i < K; i++){
        for(int k = 1; k <= block[i].c; k++)
            for(int j = block[i].a; j >= block[i].h; j--)
                dp[j] |= dp[j-block[i].h];
    }
    int ans = 0;
    for(int i = block[K-1].a; i >= 0; i--)
        if(dp[i]) {
            ans = i;
            break;
        }
    printf("%d\n", ans);
    return 0;
}

Comments