結果

問題 No.3050 Prefix Removal
コンテスト
ユーザー Andrew8128
提出日時 2025-03-07 21:58:41
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 653 bytes
コンパイル時間 1,282 ms
コンパイル使用メモリ 110,112 KB
実行使用メモリ 13,332 KB
最終ジャッジ日時 2025-03-07 21:58:54
合計ジャッジ時間 12,310 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 42 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>
using namespace std;
typedef long long ll;
int main()
{
  int N, K;
  cin >> N >> K;
  vector<ll> A(N);
  for (int i = 0; i < N; i++)
  {
    cin >> A[i];
  }
  vector<ll> c;
  ll sum = 0;
  for (int i = ssize(A) - 1; i >= 0; i--)
  {
    sum += A[i];
    if (K <= i && sum < 0)
    {
      sum = 0;
      c.clear();
    }
    else if (i != 0)
    {
      c.push_back(sum);
    }
  }
  ranges::sort(c, greater<ll>());
  cout << sum + accumulate(c.begin(), c.begin() + K-1, 0LL) << '\n';
  return 0;
}
/*
  File : ~/yukicoder/526/C.cpp
  Date : 2025/03/07
  Time : 21:45:15
*/
0