[BOJ/백준-Python]9375번 - 패션왕 신해빈

9375번 - 패션왕 신해빈

문제

해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?

입력

첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.

각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다. 모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

출력

각 테스트 케이스에 대해 해빈이가 알몸이 아닌 상태로 의상을 입을 수 있는 경우를 출력하시오.

예제 입력 1

2
3
hat headgear
sunglasses eyewear
turban headgear
3
mask face
sunglasses face
makeup face

예제 출력 1

5
3

힌트

첫 번째 테스트 케이스는 headgear에 해당하는 의상이 hat, turban이며 eyewear에 해당하는 의상이 sunglasses이므로 (hat), (turban), (sunglasses), (hat,sunglasses), (turban,sunglasses)로 총 5가지 이다.

해결할 방법을 생각해보자.💡

일반적인 조합을 알고 있다면 쉽게 풀 수 있다. 힌트를 보면 쉽게 알 수 있다. 다른 예를 들어서 설명해보겠다. A가 2개 B가 1개가 있다고 해보자. 이들이 각각 혼자서도 쓰일 수 있고 조합도 할 수 있다면 다음과 같은 조합들이 나올 것이다.

A1
A2
B1
A1, B1
A2, B2

이렇게 5개가 된다. 즉 (2개 + 1개) * (1개 + 1개) - 1개로 5개가 된 것이다. 만약에 상의 3벌과 하의 2벌이 주어진다면, (3+1) * (2+1) - 1로 11개의 경우가 나올 수 있는 것이다.

푸는 법이 알았다면 이제 예제를 잘 살펴보자.

입력 받는 문장을 공백으로 구분해서 뒤에 것이 의상의 종류가 된다.

즉 공백을 기준으로 가른 후에 그 뒤에 값들을 세주면 되는 것인데, 딕셔너리를 이용해서 기존에 존재하지 않는다면 키값으로 해당 의상종류를 넣어서 딕셔너리에 추가해주고 기존에 있던 키값이라면 밸류를 1씩 늘려주는 식으로 딕셔너리를 구성해주자.

그리고 마지막으로 경우를 나타내는 count라는 변수를 1을 저장해서 선언해준 후에 딕셔너리의 values를 반복해서 value를 꺼내서 1을 더해준 후 count에 곱해주어 마지막에 1을 빼고 출력하면 문제를 해결할 수 있다.

코드 💻

import sys

T = int(sys.stdin.readline())

for i in range(T):
  n = int(sys.stdin.readline())

  type = {}
  for i in range(n):
    wear = sys.stdin.readline().split()[1]
    if wear in type.keys():
      type[wear] += 1
    else:
      type[wear] = 1

  count = 1
  for i in type.values():
    count *= (i+1)

  print(count - 1)

해결 로그

문제 푼 날짜	체크
2022-08-06	✔

2022 196
2021 22

맨 위로 이동 ↑