結果

問題 No.2869 yuusaan's Knapsacks
コンテスト
ユーザー 寝癖
提出日時 2024-07-19 00:41:43
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 1,837 bytes
コンパイル時間 3,201 ms
コンパイル使用メモリ 257,140 KB
実行使用メモリ 814,644 KB
最終ジャッジ日時 2024-08-04 18:03:58
合計ジャッジ時間 11,060 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14 MLE * 1 -- * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

struct Data {
    ll v, use;
    Data* prev;
};

int main() {
    int N, M;
    cin >> N >> M;

    vector<ll> e(N), v(M), w(M);
    for (int i = 0; i < N; i++) cin >> e[i];
    for (int i = 0; i < M; i++) cin >> v[i] >> w[i];

    vector<ll> sum_v(1<<M), sum_w(1<<M);
    for (int S = 0; S < (1<<M); S++) {
        for (int i = 0; i < M; i++) {
            if (S>>i & 1) {
                sum_v[S] += v[i];
                sum_w[S] += w[i];
            }
        }
    }

    vector<vector<Data*>> dp(N+1, vector<Data*>(1<<M));
    dp[0][0] = new Data{0, 0, nullptr};
    for (int i = 0; i < N; i++) {
        for (int S = 0; S < (1<<M); S++) {
            if (!dp[i][S]) continue;
            int U = (1<<M)-1-S;
            for (int T = U; T > 0; T = (T-1)&U) {
                if (sum_w[T] > e[i]) continue;
                if (!dp[i+1][S|T] || dp[i][S]->v+sum_w[T] > dp[i+1][S|T]->v) {
                    dp[i+1][S|T] = new Data{dp[i][S]->v+sum_v[T], T, dp[i][S]};
                }
            }
            dp[i+1][S] = new Data{dp[i][S]->v, 0, dp[i][S]};
        }
    }

    ll m = 0;
    for (int S = 0; S < (1<<M); S++) {
        if (dp[N][S] && dp[N][S]->v > m) m = dp[N][S]->v;
    }
    int S = 0;
    for (int i = 0; i < (1<<M); i++) {
        if (dp[N][i] && dp[N][i]->v == m) {
            S = i;
            break;
        }
    }

    vector<vector<int>> ans(N);
    Data* p = dp[N][S];
    while (p) {
        for (int i = 0; i < M; i++) {
            if (p->use>>i & 1) ans[N-1].push_back(i+1);
        }
        p = p->prev;
        N--;
    }

    cout << m << endl;
    for (int i = 0; i < ans.size(); i++) {
        cout << ans[i].size();
        for (int j = 0; j < ans[i].size(); j++) cout << " " << ans[i][j];
        cout << endl;
    }
}
0