21-H0ngJu

[H0ngJu]

🔗 문제 링크

행복 유치원

✔️ 소요된 시간

30M

✨ 수도 코드

문제 요약
N명의 원생에 대해서 K개의 조로 나누었을 때 티셔츠 제작 비용이 가장 적게 들도록 하라@! 단, 조는 이웃한 아이들끼리 구성해야한다. (제작 비용은 같은 조에서 가장 큰 아이 - 가장 작은 아이이다.)

이 문제를 푸는 키는
조에 누가 있는지는 고려할 필요가 없다 인 것 같습니다.

몇명이 조인지, 누가 조인지는 고려할 필요 없이,

키큰 아이 - 키 작은 아이를 뺐을 때 가장 큰 K-1개를 제외시켜주면 자연스레 조가 나뉘게 됩니다.

예를 들어,
N = 5, K = 3이고

원생들의 키는
1 3 5 6 10 이라고 가정해봅시다.

그렇다면 이웃한 아이들끼리의 키 차이는
2 2 1 4가 됩니다.

이것을 오름차순으로 정렬하면

4 2 2 1이 되고,

조를 나누면 K-1인 2개를 빼면 되므로

2 + 1 = 3이 최소 비용이 됩니다.

수도코드

1. 이웃한 아이들끼리의 키 차이를 구한다
2. 키 차이를 내림차순으로 정렬
3. 가장 차이가 큰 K-1개를 빼고 더 해주면 최소 비용이 나온다.

저는 구현이라고 생각하고 풀었는데,
알고리즘 분류를 보니까 그리디네욤 🤔

📚 새롭게 알게된 내용

[tgyuuAn]

from heapq import *
import sys

def input(): return sys.stdin.readline().rstrip()

N, M = map(int, input().split())
children = list(map(int, input().split()))

gap = []
prev = 0
for idx, child in enumerate(children):
    if idx == 0:
        prev = child
        continue

    heappush(gap,(-1*abs(child-prev), idx))
    prev = child

link = [N]
for _ in range(M-1):
    value, idx = heappop(gap)
    heappush(link, idx)

answer = 0
prev = 0
while link:
    now_idx = heappop(link)

    if prev == now_idx-1:
        prev = now_idx
        continue

    else:
        answer += children[now_idx-1] - children[prev]
        prev = now_idx

print(answer)

헉 전 바보같이 가장 gap이 큰 아이를 빼야하겠다는 것을 우선순위큐로 구현을 했네요...

그냥 정렬로 했으면 완전 편했을텐데............................ㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠ

그래도 그래도... 원큐!!!!!!!!!!!!

[tgyuuAn]

와 그냥 diff로만 해도 되네요... 개똑똑..