[백준 13164] 행복 유치원

Junyoung Park·2022년 4월 14일
0

코딩테스트

목록 보기
362/631
post-thumbnail

1. 문제 설명

행복 유치원

2. 문제 분석

키 순서대로 정렬한 뒤, 가장 키 차이가 적게 나는 경우를 n-k번만큼 뽑는 그리디 알고리즘

3. 나의 풀이

import sys
import heapq

n, k = map(int, sys.stdin.readline().rstrip().split())
height = list(map(int, sys.stdin.readline().rstrip().split()))
height.sort()
pq = []
for i in range(n-1):
    diff = height[i+1] - height[i]
    heapq.heappush(pq, diff)
total = 0
for _ in range(n-k):
    total += heapq.heappop(pq)
print(total)
profile
JUST DO IT

0개의 댓글