프로그래머스 2단계 "위장"

문제

프로그래머스 2단계 위장

풀이

백준 9375번 문제와 같은 문제

각 옷 종류의 수 + 1(해당 옷종류를 안입는 경우)를 전부 곱해준 다음
1을 빼준 것(전부 안입는 경우)이 답이다.

Python 코드

def solution(clothes):
  dict = {}  # 각 옷 종류마다 보유 개수를 저장할 딕셔너리
  for x in clothes:
    if x[1] not in dict:
      dict[x[1]] = 1
    else:
      dict[x[1]] += 1
  
  li = list(dict.values())
  answer = 1
  for i in li:
    answer *= (i + 1) # 해당 옷 종류를 안입는 경우 추가

  return answer - 1 # 다 벗은 경우 제외