⏹️[SW Expert Academy] [Python] [D3] [1208] Flatten

문제

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&problemLevel=3&problemLevel=4&contestProbId=AV139KOaABgCFAYh&categoryId=AV139KOaABgCFAYh&categoryType=CODE&problemTitle=&orderBy=INQUERY_COUNT&selectCodeLang=ALL&select-1=4&pageSize=10&pageIndex=1

SW Expert Academy

풀이

자꾸 풀이는 하고 이상한 곳에서 오류를 범한다. 다른 사람의 코드를 확인해봐도 틀린 것이 없었는데 테스트 케이스를 확인하느라 T를 1로 지정해 놓았다. 처음 푸는 사이트다 보니 너무 헷갈리고 쓸데없는 곳에 시간을 낭비하게 된다..

 

D3 문제를 몇개 풀어보니 난이도가 한 백준 실버3-4쯤 되는 것 같다. 막 어려운 문제는 아직 안풀어보긴했지만 D4 까지는 무난하게 갈 수 있을 것 같다. 

T = 10
for test_case in range(1, T+1):
    dump = int(input())
    box = list(map(int, input().split()))
    
    for j in range(dump):
        high = max(box)
        low = min(box)

        high_index = box.index(max(box))
        low_index = box.index(min(box))
        
        box[high_index] -= 1
        box[low_index] += 1

    result = max(box) - min(box)
    print("#{} {}".format(test_case, result))

가장 높은 박스를 가장 낮은 곳으로 옮기는 것이다. 가장 높은 박스의 인덱스와 높이를 찾고 낮은 박스로 옮겨주기만 하면 된다. 중간에 0이 나오더라도 값은 동일하기 때문에 따로 break를 걸지 않았다.

 

Java 코드

package SWEA;

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

public class swea_1208 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        for(int t=1; t<=10; t++){
            int dump = Integer.parseInt(br.readLine());

            int[] arr = new int[100];
            StringTokenizer st = new StringTokenizer(br.readLine());
            for(int i=0; i<100; i++){
                arr[i] = Integer.parseInt(st.nextToken());
            }

            for(int i=0; i<dump; i++){
                Arrays.sort(arr);
                arr[0]++;
                arr[99]--;
            }
            Arrays.sort(arr);
            int result = arr[99] - arr[0];
            System.out.println("#" + t + " " + result);
        }
    }
}