結果

問題 No.1872 Dictionary Order
ユーザー ぷら
提出日時 2021-11-21 18:11:48
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 1,221 bytes
コンパイル時間 1,685 ms
コンパイル使用メモリ 199,736 KB
最終ジャッジ日時 2025-01-26 00:05:53
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

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

bitset<2001> dp[2001];

int main() {
    int N,M;
    cin >> N >> M;
    vector<int>A(N),P(N),Q(N);
    for(int i = 0; i < N; i++) {
        cin >> A[i];
    }
    for(int i = 0; i < N; i++) {
        cin >> P[i];
        Q[P[i]-1] = i;
    }
    dp[N][0] = 1;
    for(int i = N-1; i >= 0; i--) {
        dp[i] = dp[i+1]|(dp[i+1] << A[i]);
    }
    if(!dp[0][M]) {
        cout << -1 << endl;
        return 0;
    }
    vector<int>ans,tmp(N);
    ans.push_back(-1);
    for(int i = 0; i < N; i++) {
        tmp[i] = Q[i];
    }
    int now = M;
    while (now != 0) {
        vector<int>nxt;
        bool flag = false;
        for(int i = 0; i < tmp.size(); i++) {
            if(ans.back() < tmp[i] && now >= A[tmp[i]]) {
                nxt.push_back(tmp[i]);
                if(dp[tmp[i]+1][now-A[tmp[i]]] && !flag) {
                    ans.push_back(tmp[i]);
                    now -= A[tmp[i]];
                    flag = true;
                }
            }
        }
        tmp = nxt;
    }
    ans.erase(ans.begin());
    cout << ans.size() << endl;
    for(int i = 0; i < ans.size(); i++) {
        cout << ans[i]+1 << ((i+1 == ans.size())?"\n":" ");
    }
}
0