SWEA 1208 flatten (파이썬)

• 문제 링크

내 코드

# 최고점과 최저점의 차이를 반환하는 프로그램
T = 10
for tc in range(1, T +1):
    N = int(input())
    numbers = list(map(int, input().split()))
    for i in range(N):
        max_num = max(numbers)
        min_num = min(numbers)
        index_max_num = numbers.index(max_num)
        index_min_num = numbers.index(min_num)
        numbers[index_max_num] -= 1
        numbers[index_min_num] += 1
    print('#{} {}'.format(tc, max(numbers)-min(numbers)))

포인트🏹

max의 인덱스를 찾고, min의 인덱스를 찾아서
max는 -= 1, min은 += 1 해준다.
마지막에 최댓값과 최솟값의 차이를 출력해주면 끝