👨🏻💻알고리즘/프로그래머스
[프로그래머스] 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보다 작을때까지 무게별 개수를 뺴준다.