結果

問題 No.3050 Prefix Removal
ユーザー ococonomy1
提出日時 2025-03-07 21:45:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,779 bytes
コンパイル時間 2,374 ms
コンパイル使用メモリ 198,508 KB
実行使用メモリ 11,248 KB
最終ジャッジ日時 2025-03-07 21:46:08
合計ジャッジ時間 8,002 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 4 WA * 49 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#pragma GCC target("avx2")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
using pli = pair<ll,int>;
using vi = vector<int>;
using vl = vector<long long>;
#define TEST cerr << "TEST" << endl
#define AMARI 998244353
//#define AMARI 1000000007
#define el '\n'
#define El '\n'
#define YESNO(x) ((x) ? "Yes" : "No")
#define YES YESNO(true)
#define NO YESNO(false)
#define VEC_UNIQ(v) {sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end());}
#define REV_PRIORITY_QUEUE(tp) priority_queue<tp,vector<tp>,greater<tp>>
#define NO_ANS(x) {cout << (x) << '\n'; return;}

#define MULTI_TEST_CASE false
void solve(void){
    //問題を見たらまず「この問題設定から言えること」をいっぱい言う
    //一個回答に繋がりそうな解法が見えても、実装や細かい詰めに時間がかかりそうなら別の方針を考えてみる
    //添え字回りで面倒になりそうなときは楽になる言い換えを実装の前にじっくり考える
    //ある程度考察しても全然取っ掛かりが見えないときは実験をしてみる
    //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く
    //g++ -D_GLIBCXX_DEBUG -O2 c.cpp -o o
    int n;
    cin >> n;
    ll k; cin >> k;
    //abs(a_i) <= 1e7
    vector<ll> a(n);
    for(int i = 0; i < n; i++)cin >> a[i];
    
    //1 個ずつ取る場合に比べて、(N - K - 1)個余計に取れる (K == N-1) の時0個
    deque<ll> deq;
    ll deq_sum = 0LL;
    for(int i = 0; i < n - k; i++){
        deq.push_back(a[i]);
        deq_sum += a[i];
    }
    
    int idx = n - k;
    ll ans = 0LL;

    //添え字を合わせたいので
    for(ll i = 1; i <= k; i++){
        //cerr << deq_sum << ' ' << ans << el;
        bool popped = false;
        while((int)deq.size() >= 1 && deq_sum <= 0){
            ans += i * deq.front();
            deq_sum -= deq.front();
            deq.pop_front();
            popped = true;
        }
        if(!popped){
            ans += i * deq.front();
            deq_sum -= deq.front();
            deq.pop_front();
        }
        if(idx != n){
            deq.push_back(a[idx]);
            deq_sum += a[idx];
            idx++;
        }
    }
    while(!deq.empty() && deq_sum >= 0){
        ans += deq.front() * k;
        deq.pop_front();
    }
    cout << ans << el;

    return;
}

void calc(void){
    return;
}

signed main(void){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    calc();
    int t = 1;
    if(MULTI_TEST_CASE)cin >> t;
    while(t--){
        solve();
    }
    return 0;
}
0