프로그래머스 [lv2] 메뉴 리뉴얼 파이썬

코딩테스트 연습 - 메뉴 리뉴얼

 

 

 

💻  코드 

from itertools import combinations
from collections import Counter


def solution(orders, course):
    answer = []
    for n in course:  # 얘가 먼저 나와야 하는 이유는, 갯수별 최강자를 먼저 골라내야하기 때문
        temp = []
        for order in orders:
            for combination in combinations(order, n):
                temp.append(''.join(sorted(combination)))

        menu = Counter(temp)
        try:
            maximum = max(Counter(temp).values())
        except ValueError:
            pass

        for key, value in menu.items():
            if value == maximum and maximum > 1:
                answer.append(key)
    answer.sort()
    return answer

 

✅  키 포인트

1. combinations() 와 Counter() 라이브러리를 사용한다. (물론 다른 풀이도 존재하므로 필수는 아니다)

2. course 내부 원소들을 기준으로 orders 를 선별해야한다. 즉 모든 갯수마다 분류하는 것이 아니고, course 내부 원소를 기준으로 접근한다.

3. 각 조합마다 오름차순 정렬을 해주어야 한다. 그렇지 않으면 개별 케이스가 생겨버려 제대로 카운팅을 할 수 없다.

4. 두번 이상 선택된 메뉴만 결과값에 들어가는 조건을 잊지말자.