[문자열] 메뉴 리뉴얼

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

from collections import defaultdict
from itertools import combinations
 
def solution(orders, course):
    answer = []
    d = defaultdict(int)
 
    for n in course:
        for order in orders:
            temp = list(map(list, combinations(order, n)))
            for i in temp:
                i.sort()  ## WX -> WX, XW -> WX
                k = ''.join(i)
                d[k] += 1
 
        result = sorted(d.items(), key=lambda x: -x[1])
 
        if len(result) >= 1:
            max_count = result[0][1]
            if max_count >= 2:
                for cou, count in result:
                    if count == max_count:
                        answer.append(cou)
 
        d = defaultdict(int)
    return sorted(answer)
 
 
orders = ["XYZ", "XWY", "WXA"]
course = [2, 3, 5]
print(solution(orders, course))

---

문제에 대한 힌트는 https://tech.kakao.com/2021/01/25/2021-kakao-recruitment-round-1/ 에서 확인 가능하다.

유의할 점:

line 12
dict에 문자열을 추가하면 'WX'와 'XW'가 서로 다른 key로 들어갈 수 있다. 이를 방지하기 위해 문자열을 sort하고 dict에 추가했다.

line 18
5개의 메뉴가 있는 코스를 만들 수 없는 orders가 주어졌다.  이 때 result는 빈 list인데, 존재하지 않는 result의 인덱스를 참조하면 예외가 발생할 수 있다. 그래서 len(result) >= 1 이라는 if문을 추가했다.