9375번: 패션왕 신해빈
첫 번째 테스트 케이스는 headgear에 해당하는 의상이 hat, turban이며 eyewear에 해당하는 의상이 sunglasses이므로 (hat), (turban), (sunglasses), (hat,sunglasses), (turban,sunglasses)로 총 5가지 이다.
www.acmicpc.net
9375번 : 패션왕 신해빈
해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?
입력
첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.
- 각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
- 다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.
모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.
출력
각 테스트 케이스에 대해 해빈이가 알몸이 아닌 상태로 의상을 입을 수 있는 경우를 출력하시오.
생각해 볼 점
딕셔너리 타입의 자료구조, C++의 경우에는 map이 필요합니다.
입력을 받았을 때, 의류의 종류만 받아서 map에 저장합니다.
map을 <string, int>를 저장하게 하여 의류의 갯수가 몇 개인지 해시의 형태로 저장할 수 있습니다.
예를 들어
hat headgear
hat turban
의 입력을 받았다면, map에는 <hat, 2>의 형태로 저장될 것입니다.
이제 옷의 종류에 따라 갯수를 알았다면 조합의 갯수를 구하면 됩니다.
그런데, 해당 의류를 안 입었을 수도 있으므로 의류를 안 입었을 경우를 추가하여 계산해주면 됩니다.
의류의 갯수가
{hat : 3, T-shirt : 3, shoes : 4}일 경우 4 * 4 * 5 = 80개의 경우가 있습니다.
다만, 문제에서 알몸(입력된 의류를 아무것도 걸치지 않음)은 배제한다고 하였으니 80 - 1 = 79가 정답이 됩니다.
코드
#include <iostream>
#include <string>
#include <map>
using namespace std;
int main()
{
int T;
cin >> T;
for(int i = 0; i < T; i++)
{
int N;
cin >> N;
map<string, int> dic; //<종류, 갯수>
map<string, int>::iterator it;
for(int j = 0; j < N; j++)
{
string kinds, name;
cin >> name >> kinds;
it = dic.find(kinds);
if(it != dic.end()) it->second++; //이미 옷의 종류가 있으면, 갯수 + 1
else dic.insert({kinds, 1});
}
int result = 1;
for(it = dic.begin(); it != dic.end(); it++)
{
result *= it->second + 1; //입지 않은 경우를 추가하여 경우의 수 계산
}
result--; //알몸인 상태를 제외
cout << result << "\n";
}
return 0;
}
그 외
만약 입지 않는다는 경우를 생각하지 않고, 모든 경우의 수를 구하려 하면 재귀함수를 만들게 됩니다.
시도해보니 만드는 데 시간도 오래 걸리고, 덤으로 시간 초과도 받을 수 있으니 주의하세요.
'공부 및 정리 > 백준 코드' 카테고리의 다른 글
[C++]백준 - 2004번 문제 (0) | 2021.06.10 |
---|---|
[C++]백준 - 1676번 문제 (0) | 2021.06.10 |
[C++]백준 - 1010번 문제 (0) | 2021.06.04 |
[C++]백준 - 11051번 문제 (0) | 2021.06.04 |
[C++]백준 - 11050번 문제 (0) | 2021.06.04 |