[Python] 연습문제 : 귤 고르기

귤..🍅 맛있겠다

 

---

 

문제

 

---

코드

 

defaultdict 자료구조를 이용하였다. dictionary인데 없는 key를 호출하면 에러가 아닌 default 값을 뱉어준다

여기서는 defaultdict(int) 를 써서  없는 key를 호출시에 {key:0}이 생성된다

 

딕셔너리.items() 함수를 호출하면 (key, value) 가 반환이 된다

 

# tangerine 배열에서 k개를 고를 때
# 서로 다른 종류의 수의 최소값을 return
from collections import defaultdict

def solution(k, tangerine):
    minus_cnt = len(tangerine) - k
  
    ddict = defaultdict(int)

    for n in tangerine:
        ddict[n] += 1

    reduce_lst = sorted([(cnt, num) for num, cnt in ddict.items() if cnt > 0])

    while minus_cnt != 0:
        cnt, num = reduce_lst[0]
        if cnt <= minus_cnt:
            reduce_lst.pop(0)
            minus_cnt -= cnt
        else:
            minus_cnt = 0
            break

    answer = len(reduce_lst)
    

    return answer

 

 

 

 

 

---