문제

레스토랑을 운영하던 스카피는 코로나19로 인한 불경기를 극복하고자 메뉴를 새로 구성하려고 고민하고 있습니다.
기존에는 단품으로만 제공하던 메뉴를 조합해서 코스요리 형태로 재구성해서 새로운 메뉴를 제공하기로 결정했습니다. 어떤 단품메뉴들을 조합해서 코스요리 메뉴로 구성하면 좋을 지 고민하던 "스카피"는 이전에 각 손님들이 주문할 때 가장 많이 함께 주문한 단품메뉴들을 코스요리 메뉴로 구성하기로 했습니다.
단, 코스요리 메뉴는 최소 2가지 이상의 단품메뉴로 구성하려고 합니다. 또한, 최소 2명 이상의 손님으로부터 주문된 단품메뉴 조합에 대해서만 코스요리 메뉴 후보에 포함하기로 했습니다.

예를 들어, 손님 6명이 주문한 단품메뉴들의 조합이 다음과 같다면,
(각 손님은 단품메뉴를 2개 이상 주문해야 하며, 각 단품메뉴는 A ~ Z의 알파벳 대문자로 표기합니다.)

가장 많이 함께 주문된 단품메뉴 조합에 따라 "스카피"가 만들게 될 코스요리 메뉴 구성 후보는 다음과 같습니다.

각 손님들이 주문한 단품메뉴들이 문자열 형식으로 담긴 배열 orders, "스카피"가 추가하고 싶어하는 코스요리를 구성하는 단품메뉴들의 갯수가 담긴 배열 course가 매개변수로 주어질 때, "스카피"가 새로 추가하게 될 코스요리의 메뉴 구성을 문자열 형태로 배열에 담아 return 하도록 solution 함수를 완성해 주세요.

제한 사항

풀이

아이디어

• orders 배열이나 course 배열이나 최대 숫자가 작아서 combinations 함수를 써도 되겠다고 생각했다.
• orders에 있는 주문 하나하나마다 list로 변환시켜서 각각 단품메뉴로 쪼갠 배열을 만든다.
  • ex) ["A", "B", "C"]
  • 이때 이 배열은 한 번 sort를 해줘야 나중에 조합을 했을 때 서로 다른 조합으로 인식되는 경우를 방지할 수 있다. ("WX"나 "XW"나 같은 메뉴조합인데 정렬 안해줄 경우 각각 카운팅됨)
• itertools의 permutation을 사용해서 course에 있는 길이만큼씩 조합을 만들어낸다.
  • 해당 조합을 "".join()을 사용해서 하나의 문자열로 합쳐주고, defaultdict(int)에다가 1씩 늘려준다.
  • temp_dict라는 곳에는 특정 길이의 메뉴 중에서 가장 많이 언급된 메뉴의 언급횟수를 담아 줄 것이다. (특정 길이의 메뉴 중에서 가장 많이 언급된 메뉴를 골라야 하므로)
  • 그러면 각 조합이 각각 몇 번씩 등장했는지에 대한 딕셔너리가 완성되어 있을 것이다.
• 마지막으로 defaultdict.items()를 돌면서, 카운트가 2 이상인 것과(왜냐하면 문제에서 최소한 2명 이상의 손님에게서 선택받아야 한다고 했으므로), 그리고 temp_dict에서 그 아이템의 길이를 키로 가지고 있는 값과 카운트가 이상이면 answer에 append한다.

코드

from itertools import combinations
from collections import defaultdict
def solution(orders, course):
    dictionary = defaultdict(int)
    temp_dict = defaultdict(int)
    answer = []
    for menu_str in orders:
        menu_list = list(menu_str)
        menu_list.sort()
        for i in course:
            comb = combinations(menu_list, i)
            for c in comb:
                lenc = len(c)
                temp = "".join(c)
                dictionary[temp] += 1
                temp_dict[lenc] = max(temp_dict[lenc], dictionary[temp])
    
    for m, cnt in dictionary.items():
        lenm = len(m)
        if temp_dict[lenm] == cnt and cnt >= 2:
            answer.append(m)
    
    answer.sort()
    return answer