본문 바로가기
공부 정리/백준

[백준] 자바 9375 패션왕 신해빈

by 경적필패. 2021. 7. 6.
반응형

문제

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

입력

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

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

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

출력

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


테스트 케이스

 

초록색 입력/ 검은색 출력

 


접근

종류별로 받은 옷들을 입는 방법의 가짓 수를 구하는 문제입니다.

조건중 하나는 알몸인 상태는 있을 수 없다는 것입니다. 즉 아무것도 입지 않는 경우는 없다는 것입니다.

 

예를 들어 상의, 하의, 신발 종류의 의류가 있다고 할 때

상의 종류에 a, b, c 상의가 있고

하의 종류에 d, e, f 하의가 있고

신발 종류에 g, h 가있다고 할 때 경우의 수를 생각해 보겠습니다.

 

a, b, c 중에서 하나 고를 수 있고, d, e, f 중에서 하나 고를 수 있고, g, h에서 하나 고를 수 있는데 안 입는 선택지도 추가되어야 하므로

상의에서는 4가지 선택지가, 하의에서도 4가지 선택지가 신발에서도 3가지 선택지가 생기게 됩니다.

하지만 모두 안입는 선택지인 알몸은 안되므로 한 가지 경우를 빼서 4*4*3-1이 되게 됩니다.

 

이것을 hashmap을 이용하여 key값에 의류 종류를 저장하고 value값에 의류종류 개수를 저장하는 식으로 구현하고,

iterator를 이용하여 위 방식대로 계산해주었습니다.


코드

import java.awt.desktop.SystemEventListener;
import java.io.*;
import java.math.*;
import java.util.*;

public class Main {

	/*
 	9375 problem
	*/
	public static void main(String[] args) throws NumberFormatException, IOException {
		// TODO Auto-generated method stub
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		HashMap<String,Integer> hsm = new HashMap<>();
		
		int testcase = Integer.parseInt(br.readLine());
		
		
		for(int i=0; i<testcase; i++) {
			int num = Integer.parseInt(br.readLine());
			int result = 1;
			for(int j=0; j<num; j++) {
				StringTokenizer st = new StringTokenizer(br.readLine());
				st.nextToken();//의류 이름
				String kind = st.nextToken(); //의류 종류
				
				if(hsm.containsKey(kind)) {
					hsm.put(kind, hsm.get(kind)+1);
				}
				else {
					hsm.put(kind,1);
				}
			}
			Iterator<String> it =hsm.keySet().iterator();
			for(int k=0; k<hsm.size(); k++) {
				String key = it.next();
				result *= (hsm.get(key)+1);
			}
			bw.write(String.valueOf(result-1)+"\n");
			hsm.clear();
		}
		
		
		
		bw.flush();
		bw.close();
	}
	
}

주의

알몸은 안돼

 

반응형

댓글