기록 보관소

[Baekjoon/백준] 9375번: 패션왕 신해빈(C/C++) 본문

코딩 테스트/백준

[Baekjoon/백준] 9375번: 패션왕 신해빈(C/C++)

JongHoon 2022. 4. 2. 23:19

단계별로 풀어보기 17단계(정수론 및 조합론) 10번 문제

https://www.acmicpc.net/step/18

 

정수론 및 조합론 단계

N개의 물건 중 순서를 고려하지 않고 K개를 고르는 경우의 수, 이항 계수를 구하는 문제

www.acmicpc.net


백준 9375번: 패션왕 신해빈

https://www.acmicpc.net/problem/9375

 

9375번: 패션왕 신해빈

첫 번째 테스트 케이스는 headgear에 해당하는 의상이 hat, turban이며 eyewear에 해당하는 의상이 sunglasses이므로   (hat), (turban), (sunglasses), (hat,sunglasses), (turban,sunglasses)로 총 5가지 이다.

www.acmicpc.net


문제 설명

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


입력과 출력

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

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

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

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

힌트: 첫 번째 테스트 케이스는 headgear에 해당하는 의상이 hat, turban이며 eyewear에 해당하는 의상이 sunglasses이므로 (hat), (turban), (sunglasses), (hat,sunglasses), (turban,sunglasses)로 총 5가지 이다.


접근 방법

이번 문제는 테스트 케이스 T를 입력받아 의상 수 n개를 입력받는다. 그리고 다음 n개 줄에는 의상의 이름과 종류를 입력받아서 의상을 입을 수 있는 경우를 출력하는 문제다. (단, 아무것도 입지 않은 상태는 계산하지 않으며, 같은 의상 종류는 중복해서 입을 수 없다.)

문제를 읽어보면 사실상 의상 이름은 중요하지 않고, 중요한것은 의상 종류와 그 갯수뿐이라는 것을 알 수 있다. 그래서 C++의 경우 map이나 vector을 사용해서 의상 종류를 카운팅하고 그 의상 개수에 해당 의상을 입지 않는 경우를 하나씩 더해 모든 의상 종류에 대해 이것들을 추가하면 된다. 조금 설명이 이상한데 아래 예제를 통한 설명을 보면 좀 더 이해하기 쉬울 것 같다.

예제의 첫 테스트 케이스의 경우 종류에 따라 headgear : 2개, eyewear : 1개 로 나온다.
(2(headgear 개수) + 1(headgear 안 입음)) * (1(eyewear 개수) + 1(eyewear 안 입음)) - 1(둘 다 안입음)
이 되어 5가 나오게된다.

이러한 방식을 반복문을 통해서 계산해서 출력한다면 문제를 해결할 수 있다.


코드

#include <iostream>
#include <string>
#include <map>
using namespace std;

int main(int argc, char * argv[]) {
	int T, n;
	string clothesName, clothesTag;

	cin >> T;

	while(T--) {
		cin >> n;
		map<string, int> m;
		while (n--) {
			cin >> clothesName >> clothesTag;
			
			if (m.find(clothesTag) == m.end())
				m.insert({ clothesTag, 1 });
			else
				m[clothesTag]++;
		}

		int result = 1;
		map<string, int>::iterator temp;
		for (temp = m.begin(); temp != m.end(); temp++)
			result *= temp->second + 1;	//의상 종류의 개수 + 의상 종류를 입지 않은 경우 하나

		cout << result - 1 << endl;	//모두 입지 않은 경우 1개 빼기
	}

	return 0;
}

결과

백준 제출 결과

map을 습관적으로 전역 변수로 선언했다가 초기화가 되지 않아서 틀렸었다.