문제 풀이 및 개발 공간

[백준] 16165번 걸그룹 마스터 준석이 문제! (silver 3 본문

백준공부/java

[백준] 16165번 걸그룹 마스터 준석이 문제! (silver 3

gomduri43 2022. 8. 28. 21:21

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;
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));
		BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st=new StringTokenizer(br.readLine());
		HashMap<String,String> teamGroup=new HashMap<>();
		HashMap<String,String> idol=new HashMap<>();
		int n=Integer.parseInt(st.nextToken());
		int m=Integer.parseInt(st.nextToken());
		
		for(int i=0; i<n; i++) {
			String team=br.readLine();
			int numOfMember=Integer.parseInt(br.readLine());
			for(int j=0; j<numOfMember; j++) {
				String name=br.readLine();
				idol.put(name, team);
				if(teamGroup.get(team)==null) {
					teamGroup.put(team, name);
				}
				else {
					teamGroup.put(team,teamGroup.get(team).concat(","+name));
				}
			}
		}
		String[] nameOfIdol;
		for(int i=0; i<m; i++) {
			String q=br.readLine();
			int type=Integer.parseInt(br.readLine());
			if(type==1) {
				bw.write(idol.get(q)+"\n");
			}
			else {
				nameOfIdol=teamGroup.get(q).split(",");
				Arrays.sort(nameOfIdol);
				for(String e: nameOfIdol) {
					bw.write(e+"\n");
				}
			}
		}
		bw.flush();
		
	}
}