SWEA_1208 Flatten(자바) / 문제해결 기본

문제 링크

SW Expert Academy

 

package com.Expert;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Expert_1208 {

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st ;
		StringBuilder sb = new StringBuilder();
		for (int tc = 1; tc <= 10; tc++) {
			int N = Integer.parseInt(br.readLine());
			int [] arr = new int[100];
			
			st = new StringTokenizer(br.readLine());
			for (int i = 0; i < 100; i++) {
				arr[i] = Integer.parseInt(st.nextToken());
			}
			
			
			while(true) {
				Arrays.sort(arr);
				if(arr[0]+1>=arr[99] || N==0) {
					sb.append("#"+tc+" "+(arr[99]-arr[0])+"\n");
					break;
				}
				arr[99]--;
				arr[0]++;
				N--;
			}
		}
		System.out.println(sb);
	}
}

- 평탄화를 해주기 위해서 sort를 이용하면 맨 처음이 제일 낮고 마지막이 제일 높다는 점을 이용해서 문제를 풀이하면 쉽게 해결할 수 있습니다.