結果
問題 | No.3050 Prefix Removal |
ユーザー |
![]() |
提出日時 | 2025-03-07 23:02:54 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 367 ms / 2,000 ms |
コード長 | 1,190 bytes |
コンパイル時間 | 6,754 ms |
コンパイル使用メモリ | 335,348 KB |
実行使用メモリ | 39,168 KB |
最終ジャッジ日時 | 2025-03-07 23:03:15 |
合計ジャッジ時間 | 19,660 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 55 |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <atcoder/all> using namespace atcoder; #pragma GCC target("avx") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #define FASTIO cin.tie(0)->sync_with_stdio(0) using ll = long long; using mint = modint998244353; struct S { int cnt; ll sum; }; S op(S x, S y) { return S{x.cnt + y.cnt, x.sum + y.sum}; } S e() { return S{0, 0}; } int main() { FASTIO; int N, K; cin >> N >> K; vector<ll> A(N), B(N); for(int i = 0; i < N; i++) { cin >> A[i]; B[i] = A[i]; } for(int i = N - 1; i > 0; i--) { B[i - 1] += B[i]; } vector<int> p(N), q(N); iota(p.begin(), p.end(), 0); sort(p.begin(), p.end(), [&](int i, int j){ return B[i] > B[j]; }); for(int i = 0; i < N; i++) q[p[i]] = i; segtree<S, op, e> seg(N); for(int i = 1; i < N; i++) { seg.set(q[i], S{1, B[i]}); } ll offset = B[0], ans = LONG_MIN; for(int i = N - 1; i >= K - 1; i--) { int p = seg.max_right(0, [&](S x){ return x.cnt < K; }); // cout << seg.prod(0, p).sum << endl; ans = max(ans, offset + seg.prod(0, p).sum); offset -= K * A[i]; seg.set(q[i], e()); } cout << ans << endl; }