Kangho_Story

[백준] 9375번 패션왕 신해빈 C++ 풀이 본문

PS

[백준] 9375번 패션왕 신해빈 C++ 풀이

캉호 2024. 8. 29. 10:54
728x90
반응형

알고리즘 분류

  • 수학
  • 자료구조
  • 조합론
  • 해시를 이용한 집합과 맵

문제 설명

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


입력 설명

첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 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

아이디어

입력받은 옷의 종류가 존재한다면 개수를 증가시키고 존재하지 않는다면 새로운 종류를 추가한다.

이후 옷을 입을 수 있는 조합을 계산한다.


알고리즘

입력받은 옷의 종류가 존재한다면 개수를 증가시키고 존재하지 않는다면 새로운 종류를 추가한다.

이후 옷을 입을 수 있는 조합을 계산한다.

A종류 B종류 C종류의 옷이 존재한다면

이 옷들을 입을 수 있는 조합의 경우의 수는 (A+1) * (B+1) * (C+1) -1이 된다.

각 종류에 +1을 해주는 이유는 해당 종류의 옷을 입지 않는 경우의 수도 고려하기 위함이다.

마지막에 -1을 해주는 경우는 모든 종류의 옷을 입지 않는 즉, 알몸인 경우의 수를 제외하기 위함이다.


코드

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
int main()
{
	int T; cin>>T;
	while(T--)
	{
		int n, ans=1; cin>>n;
		vector<pair<int,string>> v;
		for(int i=0; i<n; i++)
		{
			string s;
			cin>>s>>s;
			auto it = find_if(v.begin(), v.end(),
			[&s](const auto& elem) { return elem.second == s; });
			if(v.end() != it) it->first++;
			else v.push_back({1,s});
		}
		for(auto &c : v) ans *= c.first+1;
		cout<<ans-1<<"\n";
	}
	return 0;
}

후기

문제 링크 -> https://www.acmicpc.net/problem/9375

옷의 개수가 0 즉, n=0인 경우도 테스트 케이스에 존재 가능하다는 것을 주의하자.


본 블로그의 모든 글은 개인적인 학습 내용이므로 다양한 오류가 있을 수 있습니다.

오류를 발견하신다면 해당 내용 댓글로 알려주시면 감사하겠습니다!

728x90
반응형
Comments