[프로그래머스] 메뉴 리뉴얼(python 파이썬)

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

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

 

조합, 사전 관련 문제이다.

course 수에 따른 조합을 구하여 사전에 넣는다.

    from itertools import combinations


    def solution(orders, course):
        answer = []
        for cour in course:
            menus = dict()
            temps = list()
            for order in orders:
                temps.extend(list(combinations(sorted(order), cour)))

            for temp in temps:
                key = ''.join(temp)
                if key in menus:
                    menus[key] += 1
                else:
                    menus[key] = 1

            for menu in menus:
                if max(menus.values()) > 1:
                    if menus[menu] == max(menus.values()):
                        answer.append(menu)

        answer.sort()

        return answer

    solution(["XYZ", "XWY", "WXA"], [2, 3, 4])