[프로그래머스] 귤 고르기 - Python

문제

---

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

프로그래머스

 

• 소요 시간 : 15분
• 난이도 : LV 2

접근 방법

---

1. 종류별 개수를 센 뒤 많은 순으로 정렬한다.

ex. 1, 3, 2, 5, 4, 2, 3
1: 1개
4: 1개
2: 2개
3: 2개
5: 2개

=> [2, 2, 2, 1, 1]

 

2. 종류별 개수 리스트를 돌면서 k만큼 가져간다. 

ex. k = 6
2+2+2 => 3번째 원소까지 가져갈 수 있다.

ex. k=4
2+2 => 2번째 원소까지 가져갈 수 있다.

풀이

---

나의 풀이

from collections import defaultdict

def solution(k, tangerine):
    
    count_dict = defaultdict(int)    
    for t in tangerine:
        count_dict[t] += 1
    
    counts = list(count_dict.values())
    counts.sort(reverse=True)
    
    answer = 0
    for count in counts:
        k -= count
        answer += 1

        if k <= 0:
            break
    
    return answer

 

딕셔너리를 사용하여 개수를 세주었다. 

---

다른 사람 풀이

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

 

로직은 동일하나 Counter 클래스로 딕셔너리를 대체하였다. 

 

https://school.programmers.co.kr/learn/courses/30/lessons/138476/solution_groups?language=python3

프로그래머스

---

나의 풀이 2

from collections import Counter

def solution(k, tangerine):
        
    counts = Counter(tangerine).most_common(k)
    answer = 0
    for size, count in counts:
        k -= count
        answer += 1

        if k <= 0:
            break
    
    return answer

 

Counter 클래스의 most_common 메소드를 사용해보았다. 

 

정리

---

 

Counter 클래스에 대해 알게 되어 정리해보았다.

 

https://ggungdi-dev.tistory.com/entry/%EC%BD%94%EB%94%A9%ED%85%8C%EC%8A%A4%ED%8A%B8%EB%A5%BC-%EC%9C%84%ED%95%9C-Python3-%EC%A0%95%EB%A6%AC
 

코딩테스트를 위한 Python3 정리