[Baekjoon] 9375_패션왕 신해빈

Silver III

문제(출처: https://www.acmicpc.net/problem/9375)

---

< 패션왕 신해빈 >

 

문제 풀이 

 

HashMap에 의상 종류를 key 값, 의상 이름을 value로 저장한다. 각 key 값에 value가 몇 개인지 센 후 입지 않은 경우도 고려하여 +1을 해준 값을 모두 곱한다. 마지막에 아무것도 안 입을 경우 1가지를 빼준다.

 

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.StringTokenizer;

public class _9375_ { // 패션왕 신해빈

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

		int t = Integer.parseInt(bf.readLine());
		HashMap<String, ArrayList<String>> map = new HashMap<>();

		for (int i = 0; i < t; i++) {
			int n = Integer.parseInt(bf.readLine());
			for (int j = 0; j < n; j++) {
				st = new StringTokenizer(bf.readLine());
				String name = st.nextToken();
				String type = st.nextToken();
				if (map.containsKey(type)) {
					ArrayList<String> arr = map.get(type);
					arr.add(name);
					map.replace(type, arr);
				} else {
					ArrayList<String> arr = new ArrayList<>();
					arr.add(name);
					map.put(type, arr);
				}
			}
			int result = 1;
			for (String key : map.keySet()) {
				int size = map.get(key).size() + 1;
				result *= size;
			}
			bw.write(result - 1 + "\n");
			map.clear();
		}
		bw.flush();
	}
}

 

Main

변수)
t : 테스트 케이스 수
map : HashMap <의상 종류, ArrayList <의상 이름>>
n : 의상 수
name, type : 의상 이름, 의상 종류
result : 의상을 입을 수 있는 경우
size : 각 의상 종류에 해당하는 의상 수

 

-  테스트 케이스 수(t) 입력

- 가진 의상의 수(n) 입력

- 가진 의상의 수(n)만큼 의상 이름(name), 의상 종류(type) 입력 후 map에 추가

- map을 순환하며 각 key 값에 해당하는 value 값이 몇 개인지 구한 후 +1 한 값을 곱함

- (result -1) 출력

---

---