프로그래머스 - 메뉴 리뉴얼

문제 링크입니다 : https://programmers.co.kr/learn/courses/30/lessons/72411?language=python3 

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

 

 

 

파이썬이 확실히 편하긴 하네.. ㅋㅋㅋㅋㅋㅋ

좋긴하다.

 

 

 

 

from itertools import combinations


def solution(orders, course):
    answer = []
    
    counts = {}
    
    for order in orders:
        sorted_order = ''.join(sorted(list(order)))
        for l in range(2, len(sorted_order) + 1):
            for food in combinations(sorted_order, l):
                str_food = ''.join(food)
                
                if str_food not in counts:
                    counts[str_food] = 1
                else:
                    counts[str_food] += 1
    
    for length in course:
        temp = []
        max_count = 0
        for key, val in counts.items():
            if len(key) == length and val >= 2:
                if val > max_count:
                    temp = [key]
                    max_count = val
                elif val == max_count:
                    temp.append(key)
        answer += temp
    
    answer.sort()
    return answer

 

 

 

 

개발 환경 : vscode

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~