結果

問題 No.2026 Yet Another Knapsack Problem
ユーザー suisen
提出日時 2022-04-25 15:05:49
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,941 bytes
コンパイル時間 4,279 ms
コンパイル使用メモリ 135,856 KB
最終ジャッジ日時 2025-01-28 21:43:21
ジャッジサーバーID
(参考情報)
judge1 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 35 TLE * 7
権限があれば一括ダウンロードができます

ソースコード

diff #

// TLE Θ(N^3) 後ろからやって状態数削減、遷移は高速化せず

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

#include <algorithm>
#include <chrono>
#include <iostream>
#include <limits>
#include <vector>

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

template <typename Fun>
int64_t measure_exec_time_ms(Fun&& f, bool out = true) {
    auto start = std::chrono::system_clock::now();
    f();
    auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now() - start).count();
    if (out) {
        std::cerr << "solved in " << elapsed << " ms" << std::endl;
    }
    return elapsed;
}

int main() {
    measure_exec_time_ms(
        [] {
            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 (uint32_t i = n; i > 0; --i) {
                const uint32_t max_num = std::min(uint32_t(n), uint32_t(n) / i);
                // dp[num][sum] = max{ pd[num-p][sum-p*i]+p*v[i] | 0<=p<=c[i] } ⋃ {-∞}
                for (uint32_t num = max_num; num > 0; --num) {
                    const uint32_t pmax = std::min(uint32_t(c[i]), num);
                    for (uint32_t sum = n; sum >= num * i; --sum) {
                        for (uint32_t p = 0; p <= pmax; ++p) {
                            dp[num][sum] = std::max(dp[num][sum], dp[num - p][sum - p * i] + 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