ABC115 C - Christmas Eve를 즐겼다.





hmax, hmin의 차이를 min으로 설정하면,
나무를 재배치하는 것이 좋다고 생각했다.

다음으로 k본씩 hmax - hmin의 차이가 최소가 되는 조합을 찾아냈다.

ChristmasEve.py
n,k = map(int,input().split())
H = []
for _ in range(n):
    h = int(input())
    H.append(h)
H.sort()
ans = float("inf")
score = 0
for i in range(n-(k-1)):
    score = H[i+k-1]-H[i]
    if ans > score:
        ans = score
print(ans)

좋은 웹페이지 즐겨찾기