結果
問題 | No.2025 Select -th Submultiset |
ユーザー |
👑 ![]() |
提出日時 | 2022-07-29 22:01:36 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 138 ms / 2,000 ms |
コード長 | 1,836 bytes |
コンパイル時間 | 4,051 ms |
コンパイル使用メモリ | 138,976 KB |
最終ジャッジ日時 | 2025-01-30 15:15:38 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
#include <iostream>#include <string>#include <vector>#include <algorithm>#include <set>#include <atcoder/modint>#include <atcoder/segtree>using namespace std;using i32 = int32_t;using u32 = uint32_t;using i64 = int64_t;using u64 = uint64_t;#define rep(i,n) for(int i=0; i<(int)(n); i++)const i64 INF = 1001001001001001001;using modint = atcoder::static_modint<998244353>;namespace RQ{u64 op(u64 l, u64 r){ return l+r; }u64 e(){ return 0; }using RQ = atcoder::segtree<u64,op,e>;}int main(){int N; cin >> N;int L; cin >> L;vector<int> C(N); rep(i,N) cin >> C[i];reverse(C.begin(), C.end());vector<vector<u64>> dp(N+1, vector<u64>(L+1,0));dp[0][0] = 1;rep(d,N){RQ::RQ rq(dp[d]);rep(i,L+1) dp[d+1][i] = rq.prod(max(0,i-C[d]), i+1);}vector<vector<u64>> sumDp(N+1, vector<u64>(L+2,0));rep(d,N+1) rep(i,L+1) sumDp[d][i+1] = sumDp[d][i] + dp[d][i];rep(i,N+1){//rep(j,L+2) cout << sumDp[i][j] << " ";//cout << endl;}int Q; cin >> Q;rep(qi, Q){u64 k; cin >> k;if(dp[N][L] < k){ cout << "-1\n"; continue; }vector<int> res(N);int len = L;for(int d=N-1; d>=0; d--){//cout << "k = " << k << " , len = " << len << endl;u64 off = sumDp[d][max(0,len-C[d])];int c = (lower_bound(sumDp[d].begin(), sumDp[d].end(), k+off) - sumDp[d].begin()) - 1;k = k - sumDp[d][c] + off;c = len - c;res[N-1-d] = c;len -= c;}rep(i,N){ if(i){ cout << ' '; } cout << res[i]; } cout << '\n';}return 0;}struct ios_do_not_sync{ios_do_not_sync(){std::ios::sync_with_stdio(false);std::cin.tie(nullptr);}} ios_do_not_sync_instance;