[알고리즘] 백준 13164 파이썬 - 행복 유치원

13164번: 행복 유치원

문제

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

이렇게 나뉘어진 조들은 각자 단체 티셔츠를 맞추려고 한다. 조마다 티셔츠를 맞추는 비용은 조에서 가장 키가 큰 원생과 가장 키가 작은 원생의 키 차이만큼 든다. 최대한 비용을 아끼고 싶어 하는 태양이는 K개의 조에 대해 티셔츠 만드는 비용의 합을 최소로 하고 싶어한다. 태양이를 도와 최소의 비용을 구하자.

 

문제풀이

n,k=map(int,input().split())
arr=list(map(int,input().split()))
ans=[]
#01
for i in range(1,n):
    ans.append(arr[i]-arr[i-1])
#02
ans.sort(reverse=True)
print(sum(ans[k-1:]))

- #01 : ans라는 리스트에 arr[i]와 arr[i-1]간의 차이를 모두 구해 추가한다.

- #02 : ans를 내림차순으로 정렬했다. 즉 차이가 많이 나는 순서대로 정렬한 것이다.

           총 k개의 조를 만들 수 있기에 차이가 적은 간격을 조로 만든다.

           참고로 단일 요소의 조는 "최대-최소=0"이다. 즉 차이가 ans의 앞 부분을

           단일 요소의 조로 하는 것이 최소의 비용을 만들 수 있다.

           그래서 최종적으로 sum(ans[k-1:])을 출력해준다.