1 초 | 128 MB | 39671 | 22016 | 18519 | 55.281% |
문제
해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?
입력
첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 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
흰트: 첫 번째 테스트 케이스는 headgear에 해당하는 의상이 hat, turban이며 eyewear에 해당하는 의상이 sunglasses이므로 (hat), (turban), (sunglasses), (hat,sunglasses), (turban,sunglasses)로 총 5가지 이다.
문제 해설
이때 문제를 보면 처음에는 옷 이름 두번째는 옷의 종류가 입력이 된다. 이때 옷이 종류가 다른 것 끼리 매칭을 시켜서 나오는 경우의 수를 따져야 한다. 1번 예제를 예시로 들면
h1, eye, h2 이 있다고 치면 우선 나올 수 있는게 h1, eye, h2로 3개는 기본 장착이다. 그리고 이후에 경우를 생각해보면 h1과 h2는 같기 때문에 같이 착용이 불가능 하지만 eye는 h1, h2와 착용이 가능하다. 그러기에 2가지의 경우가 생겨난다.
이렇게 5가지의 경우가 생겨난다.
그럼 이때 옷 이름은 필요가 없으니 옷의 종류만 필요.
n개가 있는데 n개가 같으면 경우의 수는 n
3개가 있는데 2개가 같으면 (((3-2) + 1) * (2 + 1)) - 1 = 5
4개가 있는데 2개가 같으면 ( 2(1 + 1) * (2+1) ) - 1 = 11
즉 중복된 요소는 안따지고 더하여 + 1을 해주고 중복이 되지 않은 요소는 한개마다 +1 해주어야 한다.
그럼 아래와 같이 알고리즘이 나온다.
/*
<input>
2
3
hat headgear
sunglasses eyewear
turban headgear
3
mask face
sunglasses face
makeup face
*/
#include <iostream>
#include <string>
#include <map>
#include <algorithm>
using namespace std;
int main() {
int n, c = 0;
string wear;
string body;
int result;
cin >> c;
for(int i = 0; i < c; i++) {
result = 1;
map<string, int> m;
cin >> n;
for(int i = 0; i < n; i++) {
cin >> wear >> body;
if(m.find(body) != m.end()) {
m[body]++; //중복이 되었을 경우 추가만 해줌
} else {
m[body] = 1;
}
}
for(auto iter = m.begin();iter != m.end(); iter++) {
result *= (iter->second) + 1;
//총 경우의 수를 구해야 하기 때문.
}
cout << result - 1 << '\n';
}
return 0;
}
'알고리즘' 카테고리의 다른 글
알고리즘 - 플로이드-워셜(Floyd-Warshall) (1) | 2024.06.18 |
---|---|
[백준: 1339] 단어 수학 (발표자료) (2) | 2024.06.14 |
다익스트라(Dijkstra) 알고리즘 (0) | 2024.05.01 |
[백준: 6198] 옥상 정원 꾸미기 (0) | 2024.04.09 |
[백준: 7785] 회사에 있는 사람 (0) | 2024.04.08 |