結果
問題 | No.2370 He ate many cakes |
ユーザー | 👑 Nachia |
提出日時 | 2023-07-03 21:07:56 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 65 ms / 2,000 ms |
コード長 | 1,051 bytes |
コンパイル時間 | 985 ms |
コンパイル使用メモリ | 92,920 KB |
実行使用メモリ | 15,832 KB |
最終ジャッジ日時 | 2024-07-17 19:02:25 |
合計ジャッジ時間 | 2,094 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | AC | 24 ms
6,940 KB |
testcase_04 | AC | 2 ms
6,940 KB |
testcase_05 | AC | 2 ms
6,944 KB |
testcase_06 | AC | 2 ms
6,944 KB |
testcase_07 | AC | 2 ms
6,940 KB |
testcase_08 | AC | 2 ms
6,940 KB |
testcase_09 | AC | 2 ms
6,948 KB |
testcase_10 | AC | 4 ms
6,944 KB |
testcase_11 | AC | 10 ms
6,940 KB |
testcase_12 | AC | 34 ms
9,800 KB |
testcase_13 | AC | 44 ms
10,228 KB |
testcase_14 | AC | 13 ms
6,940 KB |
testcase_15 | AC | 65 ms
15,832 KB |
testcase_16 | AC | 62 ms
12,652 KB |
testcase_17 | AC | 60 ms
14,864 KB |
testcase_18 | AC | 43 ms
6,944 KB |
testcase_19 | AC | 2 ms
6,944 KB |
ソースコード
#include <iostream> #include <string> #include <vector> #include <algorithm> #include <queue> #include <atcoder/modint> using namespace std; using i32 = int; using u32 = unsigned int; using i64 = long long; using u64 = unsigned long long; #define rep(i,n) for(int i=0; i<(int)(n); i++) const i64 INF = 1001001001001001001; using Modint = atcoder::static_modint<998244353>; int main(){ int N, K; cin >> N >> K; vector<i64> A(N); rep(i,N) cin >> A[i]; i64 offset = 0; for(auto& a : A) if(a > 0){ offset += a; a = -a; } sort(A.rbegin(), A.rend()); priority_queue<pair<i64, int>> que; vector<i64> B(N+1); rep(i,N) B[i+1] = A[i]; que.push({ offset + B[1], 1 }); vector<i64> res; res.push_back(offset); while(que.size()){ if((int)res.size() > K) break; auto [p, d] = que.top(); que.pop(); res.push_back(p); if(d != N){ que.emplace(p + B[d+1] - B[d], d+1); que.emplace(p + B[d+1], d+1); } } cout << (res[K-1]) << endl; return 0; }