結果

問題 No.2026 Yet Another Knapsack Problem
ユーザー suisensuisen
提出日時 2022-04-22 02:55:46
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,870 ms / 10,000 ms
コード長 1,724 bytes
コンパイル時間 1,028 ms
コンパイル使用メモリ 87,108 KB
実行使用メモリ 52,748 KB
最終ジャッジ日時 2023-09-26 18:47:00
合計ジャッジ時間 16,076 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 3 ms
4,380 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 2 ms
4,380 KB
testcase_09 AC 3 ms
4,376 KB
testcase_10 AC 2 ms
4,376 KB
testcase_11 AC 3 ms
4,376 KB
testcase_12 AC 1 ms
4,380 KB
testcase_13 AC 3 ms
4,380 KB
testcase_14 AC 3 ms
4,384 KB
testcase_15 AC 2 ms
4,380 KB
testcase_16 AC 3 ms
4,376 KB
testcase_17 AC 2 ms
4,380 KB
testcase_18 AC 2 ms
4,380 KB
testcase_19 AC 3 ms
4,380 KB
testcase_20 AC 1 ms
4,380 KB
testcase_21 AC 2 ms
4,384 KB
testcase_22 AC 2 ms
4,380 KB
testcase_23 AC 3 ms
4,380 KB
testcase_24 AC 1 ms
4,380 KB
testcase_25 AC 2 ms
4,376 KB
testcase_26 AC 3 ms
4,384 KB
testcase_27 AC 2 ms
4,380 KB
testcase_28 AC 44 ms
5,076 KB
testcase_29 AC 49 ms
5,188 KB
testcase_30 AC 50 ms
5,176 KB
testcase_31 AC 47 ms
5,152 KB
testcase_32 AC 37 ms
4,964 KB
testcase_33 AC 53 ms
5,336 KB
testcase_34 AC 48 ms
5,164 KB
testcase_35 AC 34 ms
4,684 KB
testcase_36 AC 44 ms
5,160 KB
testcase_37 AC 46 ms
5,100 KB
testcase_38 AC 1,870 ms
52,540 KB
testcase_39 AC 1,768 ms
52,484 KB
testcase_40 AC 1,773 ms
52,424 KB
testcase_41 AC 1,866 ms
52,444 KB
testcase_42 AC 1,866 ms
52,432 KB
testcase_43 AC 1,863 ms
52,748 KB
testcase_44 AC 1,867 ms
52,420 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// TLE? Θ(N^2(logN)^2) (using segment tree)

#include <algorithm>
#include <cassert>
#include <deque>
#include <iostream>
#include <limits>
#include <vector>

#include <atcoder/segtree>

constexpr int64_t inf = std::numeric_limits<int64_t>::max() / 2;

int64_t op(int64_t x, int64_t y) {
    return std::max(x, y);
}
int64_t e() {
    return -inf;
}

int main() {
    int n;
    std::cin >> n;

    std::vector<int32_t> c(n + 1);
    std::vector<int64_t> v(n + 1);
    for (int i = 1; i <= n; ++i) {
        std::cin >> c[i] >> v[i];
    }

    std::vector dp(n + 1, std::vector<int64_t>(n + 1, -inf));
    dp[0][0] = 0;

    for (int32_t i = n; i > 0; --i) {
        const int32_t max_num = std::min(n, n / i);
        const int32_t max_sum = n;

        // dp[num][sum] = max{ pd[num-p][sum-p*i]-(num-p)*v[i] | 0<=p<=c[i] } ⋃ {-∞} + num*v[i]
        for (int32_t num = 0; num <= max_num; ++num) {
            const int32_t sum_r = num ? i - 1 : max_sum;
            for (int32_t sum = 0; sum <= sum_r; ++sum) {
                const int32_t max_p = std::min(max_num - num, (max_sum - sum) / i);

                std::vector<int64_t> init(max_p + 1);
                for (int32_t p = 0; p <= max_p; ++p) {
                    init[p] = dp[num + p][sum + i * p] - (num + p) * v[i];
                }
                atcoder::segtree<int64_t, op, e> seg(std::move(init));

                for (int32_t p = 0; p <= max_p; ++p) {
                    dp[num + p][sum + i * p] = seg.prod(std::max(0, p - c[i]), p + 1) + (num + p) * v[i];
                }
            }
        }
    }

    for (int32_t k = 1; k <= n; ++k) {
        std::cout << *std::max_element(dp[k].begin(), dp[k].end()) << '\n';
    }

    return 0;
}
0