[백준(Baekjoon)][자바(java)] 1620 : 나는야 포켓몬 마스터 이다솜 / 집합과 맵

 

https://www.acmicpc.net/problem/1620

1620번: 나는야 포켓몬 마스터 이다솜

 

문제 풀이

 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws Exception {

		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		int n = Integer.parseInt(st.nextToken()),
			m = Integer.parseInt(st.nextToken()), i;
		Map<String, Integer> set1 = new HashMap<>();
		String set2[] = new String[n + 1], str;
		for (i = 1; i <= n; ++i) {
			str = br.readLine();
			set1.put(str, i);
			set2[i] = str;
		}
		StringBuilder sb = new StringBuilder();
		char c;
		for (i = 0; i < m; ++i) {
			str = br.readLine();
			c = str.charAt(0);
			sb.append((c >= '1' && c <= '9'
					? set2[Integer.parseInt(str)]
					: set1.get(str)) + "\n");
		}
		br.close();

		System.out.println(sb.toString());
	}
}