[SWEA] #1208 Flatten

Link

Code

import java.util.Scanner;

class Solution {
    public static void main(String args[]) throws Exception {
        Scanner sc = new Scanner(System.in);
        for (int testCase = 1; testCase <= 10; testCase++) {
            int num = sc.nextInt();
            int[] height = new int[100];
            int min = 101;
            int max = -1;
            for (int i = 0; i < 100; i++) {
                int temp = sc.nextInt();
                height[temp - 1]++;
                if (temp < min) {
                    min = temp;
                }
                if (max < temp) {
                    max = temp;
                }
            }
            for (int i = 0; i < num; i++) {
                height[max - 1]--;
                height[max - 2]++;
                if (height[max - 1] <= 0) {
                    max--;
                }
                height[min - 1]--;
                height[min]++;
                if (height[min - 1] <= 0) {
                    min++;
                }
            }
            System.out.printf("#%d %d\n", testCase, max - min);
        }
    }
}

Solution