프로그래머스 - 위장

COCOBALL·2023년 4월 8일
0

알고리즘

목록 보기
15/37

위장

문제 설명

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

종류이름
얼굴동그란 안경, 검정 선글라스
상의파란색 티셔츠
하의청바지
겉옷긴 코트

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

제한사항

  • clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
  • 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
  • 같은 이름을 가진 의상은 존재하지 않습니다.
  • clothes의 모든 원소는 문자열로 이루어져 있습니다.
  • 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
  • 스파이는 하루에 최소 한 개의 의상은 입습니다.

문제 풀이

💡 문제풀이 순서
  1. 같은 종류의 옷으로 딕셔너리에 저장한다.
  2. key 값에는 옷의 종류, value에는 옷의 이름을 저장한다
  3. 딕셔너리에 각 값을 저장할 때 key에 해당하는 value가 있으면 value 값을 카운트 1을 하고 아니라면 defalut로 0으로 지정하여 사용한다.

ex) {'headgear': ['yellow_hat', 'green_turban'], 'eyewear': ['blue_sunglasses']}

  1. 딕셔너리의 각 value 에서 1을 더한 값을 answer에 곱해준다.
    1. 1을 더하는 이유는 각 옷들을 안입는 경우가 있기 때문이다.
  2. 마지막으로 구한 answer에서 아무것도 안입는 경우를 제외하고 return 해주기 위해서 answer -1을 해준다.
def solution(clothes):
    dict = {}
    answer = 1
    for key, value in clothes:
        dict[value] = dict.get(value, 0) + 1
    
    for value in dict:
        answer *= dict[value]+1
    
    return answer -1
profile
Welcome! This is cocoball world!

0개의 댓글