結果

問題 No.3050 Prefix Removal
コンテスト
ユーザー syndrome
提出日時 2025-03-07 21:59:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,734 bytes
コンパイル時間 2,825 ms
コンパイル使用メモリ 221,080 KB
実行使用メモリ 13,340 KB
最終ジャッジ日時 2025-03-07 21:59:34
合計ジャッジ時間 14,169 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10 WA * 45
権限があれば一括ダウンロードができます

ソースコード

diff # 

// (⁠◕⁠ᴗ⁠◕⁠✿⁠)

// #pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
#define rep(i, n) for (ll i = 0; i < (n); i++)
#define srep(i, s, n) for (ll i = s; i < (n); i++)
#define len(x) ((int)(x).size())
#define all(x) (x).begin(), (x).end()
using namespace std;
template<typename T> using vc = vector<T>;
template<typename T> using vv = vc<vc<T>>;
template<typename T> using vvv = vv<vc<T>>;
using vi = vc<int>;using vvi = vv<int>; using vvvi = vv<vi>;
using ll = long long;using vl = vc<ll>;using vvl = vv<ll>; using vvvl = vv<vl>;
using ld = long double; using vld = vc<ld>; using vvld = vc<vld>; using vvvld = vc<vvld>;
using uint = unsigned int;
using ull = unsigned long long;
const ld pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
// const ll mod = 1000000007;
const ll mod = 998244353;
inline bool inside(ll y, ll x, ll H, ll W) {return 0 <= (y) and (y) < (H) and 0 <= (x) and (x) < (W); }

#define debug(var)  do{std::cout << #var << " : \n";view(var);}while(0)
template<typename T> void view(T e){cout << e << endl;}
template<typename T> void view(const vc<T>& v){for(const auto& e : v){ cout << e << " "; } cout << endl;}
template<typename T> void view(const vv<T>& vv){ for(const auto& v : vv){ view(v); } }


int main(){
    int N, K;cin >> N >> K;
    vl A(N); rep(i, N) cin >> A[i];
    while (len(A) > K && A.back() < 0) A.pop_back();
    ll ans = 0;
    vl B;
    for (int i = len(A) - 1; i >= 0; i--){
        if (i){
            A[i - 1] += A[i];
            B.push_back(A[i]);
        }else ans += A[i];
    }
    sort(all(B), greater<ll>());
    rep(i, K - 1) ans += B[i];
    cout << ans << endl;
}
0