[Programmers / Level2] 귤고르기

 

https://school.programmers.co.kr/learn/courses/30/lessons/138476

프로그래머스

---

<내 코드>

def solution(k, tangerine):
    tangerine.sort()
    frequency = {}
    for num in tangerine:
        if num in frequency: frequency[num] += 1
        else: frequency[num] = 1
        
    ret_val = set((sorted(tangerine, key=lambda x: frequency[x], reverse = True))[:k])
    return (len(ret_val))

 

-> 종류별로 count할 수 있도록 딕셔너리에다가 넣어서 숫자를 세줄 수 있도록 반복문을 돌린다.

 

람다함수 활용법을 자꾸 까먹는데.. 공식문서 에 나와있는 것 처럼

key specifies a function of one argument that is used to extract a comparison key from each element in iterable (for example, key=str.lower). The default value is None (compare the elements directly).

 

key는 어떻게 sorting할지에 대한 기준을 나타낸다고 이야기한다. 그리고 lambda 함수를 사용해서 정렬할때에는 아래와 같이 해석하자

 

‘lambda x: frequency[x]는 배열의 각 원소 x에 대해 frequency 사전에서 x의 빈도수를 반환한다. 따라서 sorted는 이 빈도수를 기준으로 배열을 정렬한다.’

 

<정해코드>

import collections
def solution(k, tangerine):
    answer = 0
    cnt = collections.Counter(tangerine)

    for v in sorted(cnt.values(), reverse = True):
        k -= v
        answer += 1
        if k <= 0:
            break
    return answer

collections 에서의 Counter을 가져와서 푸는 방법이 있다. 

counter으로 가지고 온 것은 del으로 지워야 한다고 한다. 참고하자

 

 

아래는 counter을 사용했을때 매우 유용해 보이는 메소드들이다. elements, most_common, subtract

출처: https://docs.python.org/3.8/library/collections.html?highlight=counter#collections.Counter

위의 공식문서를 간략히하자면..

 

elements : Counter으로 가지고 있는 값을 iterator으로 변환 시켜주어서 list로 반환해주는 메소드

most_common: n개를 가지고 와서 자주 나오는 순으로 반환해 주는 메소드(같은 빈도인 요소가 있으면 최초로 만난 요소가 우선됨.)

subtract: 두개의 Counter클래스에서 요소를 지워주는 메소드 ('-' 써도 됨)