프로그래머스 해시 3.위장

Jamwon·2021년 4월 14일
0

알고리즘

목록 보기
3/18

문제링크

위와 같이 2차워배열이 주어지면 위장하는 스파이가 계속 다른 조합으로 옷을 입어야된다고 한다. 무조건 하나는 입어야된다! 최소 하나의 의상을 입어도된다고 한다.

Dictionary를 이용해서 입력값을 부위별로 나누어주고 갯수를 입력해준다!

그리고 그 갯수+1 을 계속해서 곱해주는데 옷이 2가지라고 하면 그 항목을 안입는 숫자 까지 포함해야되기때문에 +1을 해서 곱해준다.

마지막에 아예 안입은 1가지만 빼주면 된다!

코드

from typing import Dict


def solution(clothes):
    dictionary = {}
    gop = 1

    for i in clothes:
        if i[-1] in dictionary:
            dictionary[i[-1]] += 1
        else:
            dictionary[i[-1]] = 1
    # print(dictionary)
    for val in dictionary.values():
        gop *= val+1

    return gop-1

딕셔너리를 이용하면 간단했던 문제!

profile
한걸음씩 위로 자유롭게

0개의 댓글