結果

問題 No.3050 Prefix Removal
ユーザー ripity
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
}
0