招商银行笔试题之修塔游戏

题目描述

小招正在玩一款修塔游戏：系统中有n座高塔，每座高塔由若干个高度相同的方块堆砌而成。修塔游戏的规则为：

1. 每次从最高塔的塔尖拿走一个方块
2. 每次在最低塔的塔尖堆砌一个方块

小招每次只能完成上述两个动作中的一个动作。游戏的目标是使n座高塔中至少有k座高塔的高度相同，请问小招最少需要多少次才能完成游戏。

输入描述：

输入共两行，第一行为n和k(1≤k≤n≤200000)，第二行为n座塔的高度组成的数组 a1, a2, …an(1≤aj≤10000)。

输出描述：

输出值为最少需要多少次动作才能完成游戏。

示例1

输入

6 5
1 2 2 4 2 3

输出

3

AC代码 [ JAVA ]

public class Main {

    private static int moveBricksCount
            (int n, int k, int[] bricks, int sum) {
        int count = Integer.MAX_VALUE;
        Arrays.sort(bricks);
        int i = 0, si = 0, j;
        while (i < n) {
            j = i;
            while (j < n && bricks[j] == bricks[i]) {
                sum -= bricks[i];
                j++;
            }
            int nd = Math.max(0, k - j + i);
            if (nd == 0) {
                count = 0;
            }
            if (j >= k) {
                count = Math.min(count, i * bricks[i] - si - (i - nd));
            }
            if (n - i >= k) {
                count = Math.min(count, sum - (n - j) * bricks[i] - (n - j - nd));
            }
            count = Math.min(count, i * bricks[i] - si + sum - (n - j) * bricks[i] - (n - (j - i) - nd));
            si += bricks[i] * (j - i);
            i = j;
        }
        return count;
    }


    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt(); 
        int k = sc.nextInt(); 
        int[] bricks = new int[n];
        int sum = 0; 
        for (int i = 0; i < n; i++) {
            bricks[i] = sc.nextInt();
            sum = sum + bricks[i];
        }
        System.out.println(moveBricksCount(n, k, bricks, sum));
    }
}