[백준-1620번/Java] 나는야 포켓몬 마스터 이다솜

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

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

 

* LinkedHashMap 을 사용했다. (순서)

import java.io.*;
import java.util.*;

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 stk = new StringTokenizer(br.readLine(), " ");

		int N = Integer.parseInt(stk.nextToken());
		int M = Integer.parseInt(stk.nextToken());

		Map<String, Integer> s = new LinkedHashMap<String, Integer>();

		for (int i = 0; i < N; i++) {
			s.put(br.readLine(), i + 1);
		}

		Object[] keyArr = s.keySet().toArray();

		for (int i = 0; i < M; i++) {
			String str = br.readLine();
			if (Character.isDigit(str.charAt(0))) {
				bw.write(keyArr[Integer.parseInt(str) - 1] + "\n");
			} else {
				bw.write(s.get(str) + "\n");
			}
		}
		bw.flush();
	}

}