Algoogle

Algorithm for Programming Contest

Typical DP Contest H - ナップザック

Category: TDPC Tag: dp

H - ナップザック

問題概要


解法


色ごとにDPする.
dp[i][j] := 重さi以下でj色以下のときの価値の最大
これを色ごとに順にやっていく.
ある色について更新するとき,元のdp配列は破壊せずにとっておく.
色の種類数を無視して更新していったあと,元の配列とのmaxを取っていく.
この際更新したほうの色数は全部+1して考えることで色の種類数のつじつまが合う.

コード


(H.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
#include <bits/stdc++.h>
using namespace std;
#define repi(i,a,b) for(int i=(int)(a);i<(int)(b);i++)
#define rep(i,n) repi(i,0,n)

int N, W, C, w[128], v[128], c[128], dp[10010][60], dp2[10010][60];

int solve() {
    rep(i,50) {
        memcpy(dp, dp2, sizeof(dp));
        rep(j,N) if(c[j] == i)
            for(int k = W; k >= w[j]; k--)
                rep(l,C+1) dp[k][l] = max(dp[k][l], dp[k-w[j]][l]+v[j]);
        rep(k,W+1) rep(l,C) dp2[k][l+1] = max(dp2[k][l+1], dp[k][l]);
    }
    int ans = 0;
    rep(i,W+1) rep(j,C+1) ans = max(ans, dp2[i][j]);
    return ans;
}

void input() {
    cin >> N >> W >> C;
    rep(i,N) {
        cin >> w[i] >> v[i] >> c[i];
        c[i]--;
    }
}

int main() {
    cin.tie(0);
    ios_base::sync_with_stdio(0);
    cout << fixed << setprecision(12);
    input();
    cout << solve() << endl;
    return 0;
}

Comments