[python] 프로그래머스_카카오21_메뉴 리뉴얼 (완전탐색, combinations)

문제

 

풀이

 

itertools 에서 combinations 을 import 해서 문자열의 조합을 이용한다. combinations 를 사용하면 아래처럼 각 튜플에 하나의 조합이 각 원소별로 쪼개져서 출력되므로, 필요에 따라 후처리가 필요하다.

a = 'abcd'
b = list(combinations(a,2))
print(b)

for i in range(len(b)):
  b[i] = ''.join(b[i])
print(b)

 

문제는 order 문자열을 순회하면서 각 주문별로 모든 조합의 개수를 foodMap으로 카운트하고, 세트의 메뉴 개수별 max값을 카운트해둔다. 그리고 마지막에 course를 순회하면서 foodMap에서 value값이 maxCnt와 동일한 key만 answer에 append하고 마지막에 answer를 sort 하고 리턴한다.

 

* dictionary를 for 문을 돌면서 key와 value를 모두 얻고 싶을때는 .items()를 이용한다.

a={'a':1,'b':2,'c':3}
print(a,end='\n\n')

for key,value in a.items():
  print('key =',key,',','values =',value)

 

 

from itertools import combinations

def solution(orders, course):
    answer = []
    foodMap = [{} for _ in range(11)] 
    maxCnt = [0 for _ in range(11)]
    
    for str in orders:
        for num in range(2,len(str)+1):
            for i in combinations(sorted(str),num):
                key = ''.join(i)
                if key in foodMap[num]:
                    foodMap[num][key] +=1
                    maxCnt[num] = max(maxCnt[num],foodMap[num][key]) 
                else:
                    foodMap[num][key] = 1
                    
    for num in course:
        for key, value in foodMap[num].items():
            if value >=2 and maxCnt[num] == value:
                answer.append(key)
                
    answer=sorted(answer)
            
    
    return answer