1seul357

[SWEA] Flatten 본문

알고리즘/SWEA

[SWEA] Flatten

1seul 2021. 12. 3. 13:42
test_case = 10  # 테스트 케이스 10개
 
for tc in range(test_case):
    dump = int(input())  # 덤프 횟수 입력받기
    Height = list(map(int, input().split()))  # 상자의 높이 입력받기
    count = 0
    while count <= dump:  # dump 횟수 초과하기 전까지 반복
        max_num = -99999
        min_num = 99999
        for i in range(len(Height)):
            if Height[i] > max_num:  # 가장 높은 위치 찾기
                max_num = Height[i]
                max_index = i
            if Height[i] < min_num:  # 가장 낮은 위치 찾기
                min_num = Height[i]
                min_index = i
        if max_num - min_num < 1:
            break
        Height[max_index] -= 1  # 상자 옮기기
        Height[min_index] += 1
        count += 1
    print('#{} {}'.format(tc + 1, max_num-min_num))

'알고리즘 > SWEA' 카테고리의 다른 글

[SWEA] 숫자 카드  (0) 2021.12.03
[SWEA] 구간 합  (0) 2021.12.03
[SWEA] 부분집합의 합  (0) 2021.12.03
[SWEA] 숫자를 정렬하자  (0) 2021.12.03
[SWEA] 특별한 정렬  (0) 2021.12.03