[백준] 9375번: 패션왕 신해빈 - 파이썬

🔈 문제

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

 

📝입력

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

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

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

 

📑출력

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

 

📚 풀이

해시를 사용한 맵을 사용하여 문제를 풀 수 있다.

의상의(똑바로 읽어도 거꾸로 읽어도 의상의! 기러기 토마토 스위스 인도인 별똥별 우영우!ㅎ) 이름과 의상의 종류가 주어졌을 때, 의상의 종류가 총 몇개인지만 카운트 해준뒤 계산하면 되었다.

계산 방법은 이렇다.

만약 headgear 2개 eyewear 1개가 있다고 가정할 때,

headgrear만 착용하는 경우의 수 2개 + eyewear만 착용하는 경우의 수 1개 + 동시에 착용하는 경우의 수 2(2*1)개를 더한 5가 정답이 된다.

이를 간단히 식으로 표현하자면 headgear를 착용하지 않는 경우의 수까지 포함한 3 * eyewear를 착용하지 않는 경우의 수까지 포함한 2 에서 알몸이 아닌 상태 1 을 빼주면 된다.

(3*2)-1

 

💻 코드

import sys
input = sys.stdin.readline

for _ in range(int(input())): # 테스트케이스
    n = int(input()) # 의상의 갯수
    ans = 1
    cabinet = dict()
    for i in range(n):
        name, frac = map(str,input().rstrip().split())
        if frac not in cabinet.keys(): # 옷장에 같은 종류의 옷이 없다면
            cabinet[frac] = 1
        else: # 이미 있는 종류라면 +1
            cabinet[frac] += 1

    for k in cabinet.keys():
        ans *= cabinet[k]+1 # (종류의 갯수 + 입지 않는 경우1)를 전부 곱한다.
    print(ans-1) # 알몸인 경우를 제외 후 출력

 

🍭 시행착오

도전!

예전 스터디할 때는 실패했던 문제였는데, 이번에 보통의 취준생을 위한 코딩테스트책을 읽으면서 (정답코드는 보지 않음!) 다시 도전해보니 바로 정답을 맞추었다.

뿌듯ㅎ

(3*2)-1 이런식을 이전에는 떠올리지 못하였는데, 이번에 풀때는 바로 딱 떠올랐다. 신기하다 이런게 성장?ㅋㅋ

저 419B 짜리 코드는 dict[종류] = [이름1, 이름2, 이름3] 이런 식으로 저장하여 후에 len(dict[종류]) 이런 식으로 연산을 하였는데, 생각해보니 이름을 굳이 체크해줄 필요가 없었다. 

그래서 dict[종류] = 3 이런식으로 바꾸어 주었고 시간이 더 단축되었다.

408B 짜리 코드는 책에 나온 해답이랑 거의 일치한다ㅎㅎ

 

🔗 문제링크 패션왕 신해빈

9375번: 패션왕 신해빈