結果

問題 No.2028 Even Choice
ユーザー erbowl
提出日時 2022-08-06 05:47:25
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 134 ms / 2,000 ms
コード長 1,083 bytes
コンパイル時間 2,105 ms
コンパイル使用メモリ 201,184 KB
最終ジャッジ日時 2025-01-30 18:58:39
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

typedef long long ll;
typedef long double ld;
#include <bits/stdc++.h>
using namespace std;
#define int long long
signed main(){
    ll n,k;
    std::cin >> n>>k;
    vector<ll> a(n);
    ll tsum = 0;
    typedef pair<ll,ll> pll;
    priority_queue<pll> pq;
    for (int i = 0; i < n; i++) {
        std::cin >> a[i];
        tsum += a[i];
        if(i>1){
            pq.push({a[i],i});
        }
    }
    ll sum = 0;
    vector<bool> used(n);
    for (int i = 0; i < k-1; i++) {
        auto [ai,ind] = pq.top();pq.pop();
        sum += ai;
        used[ind]=true;
    }
    ll ans = 0;
    for (int i = 0; i < n/2&&2*i+1<n-(k-1); i++) {
        for (int j = 2*i; j < 2*i+2; j++) {
            if(used[j]){
                sum -= a[j];
                while(true){
                    auto [ai,ind] = pq.top();pq.pop();
                    if(ind<=2*i+1)continue;
                    used[ind]=true;
                    sum += ai;
                    break;
                }
            }
        }
        ans = max(ans, a[2*i+1]+sum);
    }
    std::cout << ans << std::endl;
}
0