SWEA: 1208 Flatten

문제

SW Expert Academy

 

문제 풀이

 상자의 높낮이가 주어질 때, 가장 높은 곳에 있는 상자를 가장 낮은 곳에 있는 상자에 옮겨 평탄화를 하는 문제이다. 문제에서 평탄화를 하기 횟수(dump)가 주어질 때, 평탄화 후 가장 높은 곳의 위치와 가장 낮은 곳의 위치의 차를 반환하면 된다.

 

코드

for test_case in range(1, 11):
    dump = int(input())
    box = list(map(int, input().split()))

    for _ in range(dump):
        box[box.index(max(box))] -= 1
        box[box.index(min(box))] += 1

    print('#' + str(test_case), max(box) - min(box))