[Python - 프로그래머스 Level2] 위장

코딩테스트 연습 - 위장

문제 설명

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

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

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

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

제한 사항

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

입출력 예시

풀이 과정

해쉬를 이용하는 문제였다. 위에서 풀었던 전화번호 문제는 해쉬를 굳이 이용해야 하나? 라는 생각이 들었지만 이번에는 옷의 타입별 갯수를 세어 순열을 구현하는 문제였기 때문에 딕셔너리가 필요한 문제였다. 나는 딕셔너리 중 defaultdict를 이용하여 풀어보았다. 

일반 dictionary와 defaultdictionary의 차이는 dictionary는 미리 생성되어 있지 않는 키에 접근할 수 없지만, defaultdictionary는 미리 생성되어 있지 않는 키여도 접근이 가능하고 default값을 지정할 수 있다는 장점이 있다.

clothes 로 for문을 돌려 clothes_dict에 갯수를 세어주었다. (dict - key값은 옷의 type으로 지정)

그후 clothes_dict의 value만큼 for문을 돌려 answer에 곱해주는 형식으로 순열을 구했다. 순열을 위해 answer 를 1 더해주었기 때문에 return 할때는 answer -1 을 해주었다.

 

from collections import defaultdict


def solution(clothes):
    answer = 1
    clothes_dict = defaultdict(int)
    for c_name, c_type in clothes:
        clothes_dict[c_type] += 1
    for i in clothes_dict.values():
        answer *= (i + 1)
    return answer - 1


if __name__ == "__main__":
    print(solution([["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]))
    print(solution([["crow_mask", "face"], ["blue_sunglasses", "face"], ["smoky_makeup", "face"]]))