문제
해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?
입력
첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.
각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.
모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.
출력
각 테스트 케이스에 대해 해빈이가 알몸이 아닌 상태로 의상을 입을 수 있는 경우를 출력하시오.
예제 입력
2
3
hat headgear
sunglasses eyewear
turban headgear
3
mask face
sunglasses face
makeup face
예제 출력
5
3
내가 했던 풀이 방법
input : 테스트 케이스
n : 의상의 수
index : 테스트 케이스 배열을 관리하는 index
dress_info : 하나의 의상의 이름과 종류를 저장한 배열
answer : 알몸이 아닌 상태로 의상을 입을 수 있는 경우의 수의 모음
sum : 하나의 테스트 케이스에서 알몸이 아닌 상태로 의상을 입을 수 있는 경우의 수
dress : 의상의 종류와 개수를 저장하는 Map
dress_type : 의상의 종류를 저장하는 Set
※ dress_type은 초기 알고리즘 구상 때 필요성을 느껴 구현하였으나, 제출 후 생각해보니 정말 필요없는 코드이므로, 구현하지 말아야 함. 설명 또한 Map만을 이용한 방식으로 설명함.
- sum을 1로 초기화한 뒤, 현재 index(초기값은 0)에 해당하는 input값을 number형으로 변환하여 n에 저장한다. (sum을 1로 초기화하는 이유는 후에 경우의 수를 구하기 위해 곱하기를 이용하므로 0을 해선 안 된다. index는 input을 순환할 때마다 1씩 증가시켜주어야 한다.)
- dress Map을 만들고, n만큼 반복하면서 의상의 정보를 dress_info에 공백을 기준으로 분할하여 저장해준다. 만약 dress에 dress_info[1](이하 의상의 종류)가 있지 않다면, key를 의상의 종류로 하고 value를 1로 한 요소를 dress에 추가해준다. 만약 해당 의상의 종류가 dress에 있다면, 해당 value를 get한 뒤, 1을 증가해준 값으로 바꾸어준다. (즉, dress Map에 의상 종류 - 개수를 저장해준다.)
- 만약 dress의 size가 1일 경우, 종류가 하나밖에 없으므로, 해당 종류의 개수가 입을 수 있는 경우의 수이다.
- dress size가 1 이상일 경우, dress Map을 순환하여 각각의 value를 가져온 뒤, 1을 더한 값을 sum에 곱해준다. 모든 Map을 순환한 뒤에 sum을 answer에 1을 뺀 값을 추가해준다. (각각의 value에 1을 더해주는 이유는 해당 종류를 안 입을 수도 있는 경우를 생각해주어야 하기 때문이다. 간단히 말해, eyewear가 2가지 종류가 있을 때 1번, 2번 종류 중 하나를 입을 수도 있지만 안 입을 수도 있기 때문이다. 모든 종류를 곱해준 값에서 -1을 해주는 이유는 "알몸"이 아닌 상태로 나가야 하기 때문이다. 즉, 모든 종류를 입지 않는 경우를 제외해주어야 하기 때문이다.)
- 1~4번 방법을 cases만큼 반복한다.
코드
const fs = require('fs');
let [cases, ...input] = fs.readFileSync(0, 'utf-8').toString().trim().split('\n');
let n = 0;
let index = 0;
let dress_info = [];
let answer = '';
let sum = 1;
for (let i = 0; i < Number(cases); i++) {
sum = 1;
n = Number(input[index++].trim());
let dress = new Map();
let dress_type = new Set();
for (let j = 0; j < n; j++) {
dress_info = input[index++].trim().split(' ');
dress_type.add(dress_info[1]);
if (dress.has(dress_info[1])) {
dress.set(dress_info[1], dress.get(dress_info[1]) + 1);
} else {
dress.set(dress_info[1], 1);
}
}
if (dress.size === 1) {
answer += n + '\n';
} else {
dress_type.forEach((type) => {
sum *= Number(dress.get(type) + 1);
});
answer += sum - 1 + '\n';
}
}
console.log(answer.trim());
회고
이 문제도 어렵지는 않았던 문제였지만, sum값을 초기화해주는 것을 까먹어서 한 번 틀렸던 문제 (case가 2개 이상이 되면 기존 값에 곱해져서 문제 발생) 간단한 걸 놓쳐서 실수했다는 게 아쉬웠다. 항상 꼼꼼하게 보질 않아서 틀리는 문제가 있는 것 같다. 조금만 더 문제 조건에 집중을 해서 풀어야겠다.
