둘셋 개발!

[알고리즘] 프로그래머스 메뉴 리뉴얼 본문

알고리즘

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

23 2022. 6. 8. 18:35

 

 

포스트를 작성하기 전에 '개발하는 사막여우' 블로그의 소스코드를 가져온 것을 미리 알립니다!

(참고 해당 블로그 링크 :https://dev-note-97.tistory.com/128)

 

✏️문제 링크 :

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

 

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

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

programmers.co.kr


✏️해답 sourceCode :

from itertools import combinations, permutations
from collections import Counter

def solution(orders, course):
    answer = []
    for c in course:
        temp = []
        for order in orders:
            combi = combinations(sorted(order), c) 
            temp += combi
        counter = Counter(temp) # 가장 많이 주문한 단품메뉴의 주문 횟수 
        if len(counter) != 0 and max(counter.values()) != 1: # 중복된 것이 1개라도 있으면
            
            # 한줄 :answer += [''.join(f) for f in counter if counter[f] == max(counter.values())]
            for f in counter : 
                if counter[f] == max(counter.values()) : # 가장 많이 함께 주문한 단품메뉴 찾기
                    answer+=[''.join(f)] #문자열로 전환
                    

    return sorted(answer)
1. 주문한 단품의 조합을 찾기 위해 Combinations를 사용해야 겠다고는 생각해는데
내가 고민한 부분은 만약 'ABCD' 와 'CA'가 있고 2개의 단품으로 구성된 코스를 구한다고 하면, 순열로 찾아야 단품 A와 단품 C가 중복된 것을 알 수 있을 거라 생각해서 복잡했는데
해답 소스코드를 보니 그냥 sorted하면 간단하게 해결될 문제였다....!! 이런 한 끗차이가 정말 므숩다
2. 문제를 잘못 이해하고 있었다. 가장 많이 함께 주문한 단품메뉴를 찾아야 하는데 나는 손님 2명이상이 중복된 단품 메뉴를 고르면 그것도 다 포함시켜야 하는 것으로 착각했다...문제를 똑바로 읽고 이해하자

 

✏️새로 알게 된 개념

Counter() 함수를 사용하면 중복된 횟수를 구할 수 있다.

from collections import Counter

arr = [0,0,1,1,2,3,3,3,3,4,5]
print(Counter(arr))

Counter({3: 4, 0: 2, 1: 2, 2: 1, 4: 1, 5: 1})