👨🏻‍💻알고리즘/프로그래머스

[프로그래머스] Lv2. 귤 고르기

waveofmymind 2023. 1. 22. 20:23

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/138476?language=python3 

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

코드

from collections import Counter

def solution(k, tangerine):
    cnt = 0
    c = Counter(tangerine)
    arr = list(c.items())
    arr.sort(reverse=True,key = lambda x : x[1])
    for i in arr:
        if k <= 0:
            break
        k-=i[1]
        cnt+=1
    return cnt

풀이

  • Counter 모듈은 입력받은 리스트의 원소들의 개수를 세어 딕셔너리로 반환해준다.
  • 귤의 무게별 개수를 딕셔너리로 정리하고, items()를 이용해서 딕셔너리를 튜플로 된 리스트로 arr을 초기화한다.
  • 그 후, 1번 인덱스가 value, 즉 무게별 개수이므로 무게별 개수로 내림차순 정렬을 하면, 0번 인덱스가 제일 개수가 많은 무게이다.
  • 그리고 for문을 이용해서 k가 0보다 작을때까지 무게별 개수를 뺴준다.