programming language/Algorithm

[백준] 13164 행복 유치원 (python, Greedy)

jellylucy 2023. 12. 10. 11:47

https://www.acmicpc.net/problem/13164

 

13164번: 행복 유치원

행복 유치원 원장인 태양이는 어느 날 N명의 원생들을 키 순서대로 일렬로 줄 세우고, 총 K개의 조로 나누려고 한다. 각 조에는 원생이 적어도 한 명 있어야 하며, 같은 조에 속한 원생들은 서로

www.acmicpc.net

문제 풀이

1. 아이들 키 차이에 대한 리스트를 만든다.

2. 이를 sorting한다. 

3. n-k 개 만큼 선택한다.

 

결과를 보면 차이가 적은 것을 n-k 만큼 선택하면 

최솟값으로 차이를 가진 아이들의 조가 만들어진다.

n, k = map(int, input().split())
graph = list(map(int, input().split()))
lenGraph = []
for i in range(n-1):
    lenGraph.append(graph[i+1]-graph[i])
lenGraph.sort()
result = 0
for i in range(n-k):
    result += lenGraph[i]
print(result)