[코딩테스트] 프로그래머스 귤 고르기 파이썬(Python)

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

프로그래머스

from collections import Counter
def solution(k, tangerine):
    counter =  Counter(tangerine)
    sorted = counter.most_common()
    result = set()
    for num, count in sorted:
        k -= count
        result.add(num)
        if k <= 0:
            break
    return len(result)

Counter 모듈을 사용하여 tangerine 리스트의 저장되어 있는 원소의 개수를 세고 most_common() 함수를 사용하여

원소의 개수를 기준으로 내림차순으로 정렬하였습니다.

예를 들어 [1, 3, 2, 5, 4, 5, 2, 3] 형태로 입력이 들어왔다면

[(3, 2), (2, 2), (5, 2), (1, 1), (4, 1)] 형태로 정렬이 됩니다.

위의 리스트에서 num, count를 꺼내면서 k 에서 count만큼 제거해주며 결과에 해당 num을 담아줍니다.

종료조건으로는 k(귤의 개수)값이 0과 같거나 작다면 종료해줍니다.

result의 길이를 반환하여 문제를 해결합니다.