結果
問題 |
No.2025 Select $k$-th Submultiset
|
ユーザー |
![]() |
提出日時 | 2022-07-29 23:05:36 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 168 ms / 2,000 ms |
コード長 | 2,107 bytes |
コンパイル時間 | 4,591 ms |
コンパイル使用メモリ | 256,404 KB |
最終ジャッジ日時 | 2025-01-30 15:38:44 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; //using mint = modint1000000007; //const int mod = 1000000007; //using mint = modint998244353; //const int mod = 998244353; //const int INF = 1e9; //const long long LINF = 1e18; #define rep(i, n) for (int i = 0; i < (n); ++i) #define rep2(i,l,r)for(int i=(l);i<(r);++i) #define rrep(i, n) for (int i = (n-1); i >= 0; --i) #define rrep2(i,l,r)for(int i=(r-1);i>=(l);--i) #define all(x) (x).begin(),(x).end() #define allR(x) (x).rbegin(),(x).rend() #define endl "\n" #define P pair<int,int> template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; } template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; } long long op(long long a, long long b) { return a + b; } long long e() { return 0; } long long target; bool f(long long v) { return v < target; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int n, l; cin >> n >> l; vector<long long>c(n); rep(i, n)cin >> c[i]; vector< segtree<long long, op, e>>seg(n); rep(i, n) { segtree<long long, op, e>_seg(l + 1); seg[i] = _seg; } rrep(i, n) { if (i == n - 1) { rep(j, c[i] + 1) seg[i].set(j, 1); } else { rep(j, l + 1) { int mn = max(0LL, j - c[i]); int mx = j; auto get = seg[i + 1].prod(mn, mx + 1); seg[i].set(j, get); } } } /*rep(i, n) { rep(j, l + 1) { cout << seg[i].get(j) << " "; } cout << endl; }*/ int q; cin >> q; long long mx = seg[0].get(l); while (q--) { long long k; cin >> k; if (mx < k) { cout << -1 << endl; continue; } vector<int>ans(n); int sz = l; rep(i, n - 1) { //cout << "?" << endl; int mn = max(0LL, sz - c[i]); target = k; int idx = seg[i + 1].max_right<f>(mn); //cout << mn << " " << idx << endl; long long get = seg[i + 1].prod(mn, idx); k -= get; ans[i] = sz - idx; sz = idx; if (k <= 0)break; } ans[n - 1] = sz; rep(i, n)cout << ans[i] << " "; cout << endl; } return 0; }