[프로그래머스] 메뉴 리뉴얼

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

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

 

이 문제를 처음 풀었을 때 문제 이해를 잘 못해서 시간이 조금 걸렸었다.

 

쉽게 말하면 정해진 개수로 단품 요리를 만들었을 때 가장 많이 주문된 조합을 구하면 된다.

combinations 을 활용하여 정해진 개수에 대해 조합을 구하여 menuList 에 담고 Counter 를 활용하여 조합된 요리가 몇 번 나왔는지 확인한다.

 

입출력 예제에서 정해진 개수가 2일 때, 

menuList = [('AC', 4), ('CD', 3), ('CE', 3), ('DE', 3), ('BC', 2), ('BF', 2), ('BG', 2), ('CF', 2), ('CG', 2), ('FG', 2), ('AD', 2), ('AE', 2), ('AB', 1), ('AF', 1), ('AG', 1), ('AH', 1), ('CH', 1), ('DH', 1), ('EH', 1)]

'AC' 가 4번, 'CD' 가 3번, 'EH' 가 1번 주문되었다는 것을 의미한다. 

 

아래의 menuList_[0][1] 값이 가장 많은 주문 횟수이며 조건에 의해 1보다 크다면 menuList_[0][1] 의 주문 횟수를 가진 조합된 메뉴가 '스카피'가 만들게 될 코스요리 메뉴 구성 후보이다.

 

from itertools import combinations
from collections import Counter

def solution(orders, course):
    answer = []
    
    for i in course:
        menuList = []
        
        for j in orders:
            for e in combinations(j, i):
                menuList.append(''.join(sorted(e)))
                
        menuList_ = Counter(menuList).most_common()
    
        for menu, cnt in menuList_:
            if cnt > 1 and cnt == menuList_[0][1]:
                answer.append(menu)
            else:
                break
    
    answer = sorted(answer)
    
    return answer

print(solution(['ABCFG', 'AC', 'CDE', 'ACDE', 'BCFG', 'ACDEH'],	[2, 3, 4])) # ['AC', 'ACDE', 'BCFG', 'CDE']