9375번 - 패션왕 신해빈
시간제한 : 1초
메모리 제한 : 128MB
문제
해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?
입력
첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 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코드
import sys
T = int(sys.stdin.readline().strip())
# 테스트케이스 개수만큼 반복
for _ in range(T):
wears = {}
n = int(sys.stdin.readline().strip())
# 의상 개수만큼 반복
for _ in range(n):
name, type = sys.stdin.readline().strip().split()
if type in wears:
# 종류가 이미 있으면 해당 종류에 의상 이름만 추가
wears[type].append(name)
else:
# 종류가 사전형에 없으면 추가
wears[type] = [name]
cnt = 1
# 조합식을 세워 계산
# +1하는 이유는 알몸도 옷이라고 생각해 추가
for x in wears:
cnt *= (len(wears[x]) + 1)
# -1하는 이유 모든 종류의 옷을 착용하지 않았을 경우를 제외
print(cnt-1)
풀이
시간 : 44ms
메모리 : 31256KB
이 문제는 어떤 알고리즘을 사용해 풀어야할지 감을 잡지 못해 구글링해본 결과 조합 알고리즘을 이용해 푸는 방법이 있었다.
의상의 조합 수는 아래와 같이 찾을 수 있다.
예를 들면 의상의 종류가 head, top, bottom 이렇게 세 개있다고 가정하면 해당 종류의 의상을 착용하지 않은 상태도 하나의 의상이라고 생각하여 각 의상의 종류마다 +1을 해주고 의상의 종류의 개수들끼리 곱해준다.
그 결과 모든 의상의 조합이 나오게 되는데 이 조합에서 모든 의상의 종류를 하나도 입지 않은 즉, 알몸인 경우를 생각해 -1을 해준 후 출력해주면 된다.
끄적끄적