티스토리 뷰

728x90

combinations 와 counter 을 알 수 있게 해준 문제였다 !!

 

문제

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

 

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

레스토랑을 운영하던 스카피는 코로나19로 인한 불경기를 극복하고자 메뉴를 새로 구성하려고 고민하고 있습니다. 기존에는 단품으로만 제공하던 메뉴를 조합해서 코스요리 형태로 재구성해서

programmers.co.kr

 

소스코드

from itertools import combinations
from collections import Counter

def solution(orders, course):
    answer = []
    for cnt in course:
        tmp = []
        for order in orders:
            combi = combinations(sorted(order), cnt)
            tmp += combi
        count = Counter(tmp)
        if len(count)!=0 and max(count.values())!=1:
            for i in count:
                if count[i] == max(count.values()):
                    answer += [''.join(i)]
    return sorted(answer)

 

반응형