2 분 소요

[Silver III] 패션왕 신해빈 - 9375

문제 링크

성능 요약

메모리: 31120 KB, 시간: 40 ms

분류

조합론, 자료 구조, 해시를 사용한 집합과 맵, 수학

제출 일자

2024년 3월 25일 14:29:00

문제 설명

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

입력

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

  • 각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
  • 다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.

모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

출력

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

풀이

테스트케이스를 보자마자 딕셔너리를 사용해야겠다는 생각이 들었다. 각 의상의 종류를 순회하면서 경우의 수를 곱해주면 값을 구할 수 있다.

T = int(input()) # 테스트케이스 수 입력

for i in range(T):
    fassion = dict() # 의상을 저장할 딕셔너리 선언
    n = int(input()) # 의상의 수 입력

먼저 필요한 수를 입력받는다.


    for _ in range(n):
        a, b = map(str, input().rstrip().split())
        if fassion.get(b) == None: # b종류가 없다면,
            fassion[b] = set() # b에 집합자료형 형성
        fassion[b].add(a) # b종류에 a의상 추가

입력받은 의상의 수만큼 a,b변수에 저장한다. a는 의상이름, b는 의상종류이다. 만약 fassion딕셔너리에 b(의상종류)가 없다면, b라는 키값에 집합자료형 set()으로 초기화해준다. 이후 순차적으로 b에 a를 추가해준다. 


    cnt = 1 # 경우의 수 초기화
    for i in fassion.keys(): # 의상종류(key)를 순회
        # 해당 종류의 각 옷을 입는 경우(옷 개수) + 해당 종류의 옷을 입지 않는 경우(1)
        cnt *= len(fassion[i]) + 1
    print(cnt-1) # 모든 의상을 하나도 입지 않은 경우 제외

의상 종류와 이름 추가가 끝나면 fassion딕셔너리를 순회하며 경우의 수를 계산한다. .keys()를 이용해서 키값(의상종류)에 있는 의상 수에 접근한다.

  • 해당 종류의 의상을 한번씩 착용 == 의상 수
  • 해당 종류의 모든 의상을 미착용 == 1
이렇게 각 의상을 한번씩 입은 경우의 수와 하나도 입지않은 수 1을 더한 값을 곱해나간다. 마지막에는 모든 의상을 하나도 입지않은 경우를 제외해야 하므로 구한 경우의 수에서 1을 빼준다.

전체 코드

import sys

input = sys.stdin.readline

T = int(input())

for i in range(T):
    fassion = dict() # 의상을 저장할 딕셔너리 선언
    n = int(input())
    for _ in range(n):
        a, b = map(str, input().rstrip().split())
        if fassion.get(b) == None: # b종류 의상에 아무것도 없다면,
            fassion[b] = set() # b에 집합자료형 형성
        fassion[b].add(a) # b종류에 a의상 추가

    cnt = 1 # 경우의 수 초기화
    for i in fassion.keys(): # 의상종류(key)를 순회
        # 해당 종류의 각 옷을 입는 경우(옷 개수) + 해당 종류의 옷을 입지 않는 경우(1)
        cnt *= len(fassion[i]) + 1
    print(cnt-1) # 모든 의류를 하나도 입지 않은 경우 제외

댓글남기기

참고