最优清零方案（JAVA）

题目：

给定一个长度为 N 的数列 A1,A2,⋯,AN 。现在小蓝想通过若干次操作将这个数列中每个数字清零。

每次操作小蓝可以选择以下两种之一:

选择一个大于 0 的整数, 将它减去 1 ;
选择连续 K 个大于 0 的整数, 将它们各减去 1 。
小蓝最少经过几次操作可以将整个数列清零?

输入格式
输入第一行包含两个整数 N 和 K 。

第二行包含 N 个整数 A1,A2,⋯,AN 。

输出格式
输出一个整数表示答案。

样例输入

4 2
1 2 3 4

样例输出

6

评测用例规模与约定
对于 20 %20% 的评测用例,1≤K≤N≤10 。

对于 40 %40% 的评测用例, 1≤K≤N≤100 。

对于 50 %50% 的评测用例, 1≤K≤N≤1000 。

对于 60 %60% 的评测用例, 1≤K≤N≤10000 。

对于 70 %70% 的评测用例, 1≤K≤N≤100000 。

对于所有评测用例, 1≤K≤N≤1000000,0≤Ai≤1000000 。

运行限制
最大运行时间：15s
最大运行内存: 512M

代码：

import java.util.Scanner;
 
import java.util.Arrays;
// 1:无需package
// 2: 类名必须Main, 不可修改
 
public class Main {
    static long count = 0;
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int k = scanner.nextInt();
 
        long [] A = new long[n];
        for (int i =0 ; i < n;i++){
            A[i] = scanner.nextLong();
        }
        f(A,k);
        for (int i=0;i<n;i++){
            count+=A[i];
        }                           //循环后都是不连续的数字只能一个个减
        System.out.println(count);
    }
 
    public  static void f(long[] A,  int k){
        int m =0;
        while (m <= A.length-k){                  //从第一个数字开始将k个连续数字减去其中最小值，依次向后进行。
            long min =Integer.MAX_VALUE;
            int index = -1;
            for (int i =m ;i<m+k ;i++ ){
                if (A[i]<=min){
                    min =A[i];
                    index = i;
                }
            }
            for (int i =m ;i<m+k ;i++ ){
                A[i] -= min;
            }
            count+=min;          //最小数字则为对连续k个数字的操作次数。
            m =index+1;         //向后移动，移动到0的下一位
 
        }
 
 
 
 
    }
}