문제 링크

https://programmers.co.kr/learn/courses/30/lessons/72411

문제 설명

• 손님들의 주문 리스트가 주어진다
• 코스로 만드려는 메뉴 개수가 주어진다
• 개수별로 가장 많이 나온 조합을 리턴한다

풀이

• itertools 의 combinations 함수를 이용해 딕셔너리에 각 조합의 개수를 카운트
• 가장 많이 나온 조합을 리스트에 append 해서 리턴한다

느낀 점

• 혹시 몰라서 다른 사람의 풀이를 봤는데 collections 의 Counter 라는 편리한 클래스가 있었다

코드

from itertools import combinations


def solution(orders, course):
    answer = []
    for i in course:
        count = dict()
        for order in orders:
            for combi in combinations(order, i):
                combi = tuple(sorted(combi))
                if combi not in count:
                    count[combi] = 0
                count[combi] += 1
        if not count:
            continue
        max_count = max(count.values())
        if max_count < 2:
            continue
        for key in count:
            if count[key] == max_count:
                answer.append(''.join(key))
    answer.sort()
    return answer