[백준] 16165. 걸그룹 마스터 준석이 (Java)

 

문제

16165번: 걸그룹 마스터 준석이

 

알고리즘

해시맵

 

풀이

팀 이름을 key로, 멤버의 리스트를 value로 저장하는 [팀-멤버 리스트] 해시맵과

멤버 이름을 key로, 팀 이름을 value로 저장하는 [멤버-팀] 해시맵을 만들어

퀴즈의 종류에 따라 적절한 해시맵을 선택해 value를 출력한다.

 

* 멤버의 이름을 사전순으로 출력해야하므로 리스트를 정렬한 후 해시맵에 삽입한다.

 

코드

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		StringTokenizer st = new StringTokenizer(br.readLine(), " ");
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());

		HashMap<String, ArrayList<String>> teamMember = new HashMap<>();
		HashMap<String, String> memberTeam = new HashMap<>();

		for (int i = 0; i < N; i++) {
			String team = br.readLine();
			int num = Integer.parseInt(br.readLine());
			ArrayList<String> memberList = new ArrayList<>();
			for (int j = 0; j < num; j++) {
				String member = br.readLine();
				memberList.add(member);
				memberTeam.put(member, team);
			}

			Collections.sort(memberList);
			teamMember.put(team, memberList);
		}

		StringBuilder sb = new StringBuilder();

		for (int i = 0; i < M; i++) {
			String name = br.readLine();
			int type = Integer.parseInt(br.readLine());

			if (type == 0) {
				for (String member : teamMember.get(name))
					sb.append(member).append("\n");
			} else {
				sb.append(memberTeam.get(name)).append("\n");
			}
		}

		System.out.println(sb);
	}
}