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

사용 알고리즘: 해시

사용 언어: java

 

package week33;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

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 k = Integer.parseInt(st.nextToken());
		Map<String, Integer> nameToNum = new HashMap<>();
		Map<Integer, String> numToName = new HashMap<>();
		
		for (int i = 1; i <= n; i++) {
			String s = br.readLine();
			nameToNum.put(s, i);
			numToName.put(i, s);
		}
		
		for (int i = 0; i < k; i++) {
			String s = br.readLine();
			int flag = s.charAt(0) - '0';
			if(0<=flag && flag<=9) {
				System.out.println(numToName.get(Integer.parseInt(s)));
			} else {
				System.out.println(nameToNum.get(s));
			}
		}
	}

}

 

 

어렵지 않게 아이디어를 떠올릴 수 있는 문제.

Scanner를 사용하면 뭔가 입력이 잘 안됨.. 

BufferedReader를 사용하는 것이 마음 편함.