题解 | 小红闯关

贪心

import java.util.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n=in.nextInt();
        int k=in.nextInt();
        int[] a=new int[n];
        long totalTime=0;
        for(int i=0;i<n;i++){
            a[i]=in.nextInt();
            totalTime+=a[i];
        }
        PriorityQueue<Integer> pq=new PriorityQueue<>(Collections.reverseOrder());
        long savedTime=0;
        for(int i=n-1;i>=0;i--){
            if((i+1)%k==0&&!pq.isEmpty()){
                savedTime+=pq.poll();
            }
            pq.add(a[i]);
        }
        System.out.println(totalTime-savedTime);

    }
}