题解 | 小红闯关

#include<bits/stdc++.h>
#include <queue>
using namespace std;
int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    int n,k;
    cin >> n >> k;
    vector<int> a(n);
    for(int i=0; i<n; i++){
        cin >> a[i];
    }
    priority_queue<int> maxh;
    long long sum=0;
    for(int i=n-1; i>=0; i--){
        sum+=a[i];
        maxh.push(a[i]);
        if(i!=0 && (i)%k==0){
            sum-=maxh.top();
            maxh.pop();
        }
    }
    cout << sum;
}