문제
해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?
입력
첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.
- 각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
- 다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.
모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.
출력
각 테스트 케이스에 대해 해빈이가 알몸이 아닌 상태로 의상을 입을 수 있는 경우를 출력하시오.
이 문제는 조합문제였습니다. 결국 착용하지 않는 의상이 있을 수 있는데 하나도 안 입는경우는 없을때의 모든 조합의 수를 말하고 있는 것입니다. 조합의 경우를 다 출력하라는 것이 아니라 갯수를 파악하라고 한 것이라면 더 쉽게 해결할 수 있습니다.
예를 하나 들어서 설명해보겠습니다.
보이는 것과 같이 걸치지 않는 것도 하나의 경우로 생각하고서 조합을 계산하되, 모두 걸치지 않는 경우는 존재하지 않는다는 문제의 조건을 고려해서 모든 종류에서 None일때인 경우의 수 1개를 빼준 것이 문제의 답이 됩니다.
문제를 쉽게 해결하기 위해서 Dict 자료형을 활용했습니다.
코드는 다음과 같습니다
T = int(input())
dic = {}
ans = []
for _ in range(T):
C = int(input())
for i in range(C):
item, category = input().split()
if category in dic:
dic[category] += 1
else:
dic[category] = 2 # 카테고리 첫 등록 시 item + None으로 2기지 경우의 수가 들어가짐
tmp = 1
if len(dic.values()) == 0: # 가진 의상의 수가 0<= n <= 30을 고려
ans.append(0)
else:
for j in dic.values():
tmp *= j
ans.append(tmp-1) # 모든 경우의 수에서 모두 None 일 경우의 수 1개를 감산
dic = {} # 새로운 테스트 케이스를 시작하기 전 초기화
for i in ans:
print(i)
'알고리즘 > 문제풀이' 카테고리의 다른 글
백준 11726번 2XN 타일링 - Python (0) | 2023.04.02 |
---|---|
백준 9461번 파도반 수열 - Python (0) | 2023.04.01 |
백준 11286번 절대값 힙 - Python (0) | 2023.04.01 |
백준 11279번 최대 힙 - Python (0) | 2023.03.22 |
백준 9095번 1,2,3 더하기 - Python (0) | 2023.03.19 |