본문 바로가기

코딩테스트 with PYTHON

[python] 코딩테스트 대비 - 그리디(시뮬레이션-높이 맞추기)

문제)

 

답안코드)

1
2
3
4
5
6
7
8
9
10
11
12
import sys
#sys.stdin=open("input.txt""r")
L=int(input())
a=list(map(int, input().split()))
m=int(input())
a.sort()#정렬 인덱스 0이 가장 높은 상태
for _ in range(m):#m번 반복
    a[0]+=1#최대 값 1 빼고
    a[L-1]-=1#최소 값 1 더하기
   a.sort()#그 후 재 정렬
 
print(a[L-1]-a[0])#답 출력
cs