ABC81 C - Not so Diverse를 풀었다






각 요소를 세어 본다.
K 종류 이하가 될 때까지 요소수가 적은 순서로 더하면 답이 될 것이라고 생각했다.

NotSoDiverse.py
N,K = map(int,input().split())
A = list(map(int,input().split()))

dic = {}
for n in range(N):
    if A[n] not in dic:
        dic[A[n]] = 0
    dic[A[n]] += 1

target = len(dic)-K
dic = sorted(dic.items(), key=lambda t:t[1])
#print(dic)
ans = 0
for i in range(target):
    ans += dic[i][1]
print(ans)

사전의 sort가 항상 잊어버린다.

신 사이트에 감사

좋은 웹페이지 즐겨찾기