백준 1620. 나는야 포켓몬 마스터 이다솜

문제:

 

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

 

나의 첫 번째 풀이(시간초과) - Map 활용

package test;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.Map;
import java.util.StringTokenizer;

public class test67 {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		try {
			HashMap<Integer, String> pokemons = new HashMap<>();
			StringTokenizer st = new StringTokenizer(br.readLine());
			int n = Integer.parseInt(st.nextToken());
			int m = Integer.parseInt(st.nextToken());
			for (int i = 1; i <= n; i++) {
				int key = i;
				String name = br.readLine();
				pokemons.put(key, name);
			}
			for (int i = 0; i < m; i++) {
				String lookFor = br.readLine();
				if (lookFor.matches("\\d+")) {
					int key = Integer.parseInt(lookFor);
					bw.write(pokemons.get(key));
				} else {
					for (Map.Entry<Integer, String> entry : pokemons.entrySet()) {
						if (entry.getValue().equals(lookFor)) {
							bw.write(String.valueOf(entry.getKey()));
						}
					}

				}
				if (i < m - 1) {
					bw.write("\n");
				}
			}
			bw.flush();
			bw.close();
			br.close();
		} catch (NumberFormatException | IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

}

 

나의 두번째 풀이 - ArrayList와 Map을 동시에 사용(정답)

package test;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.StringTokenizer;

public class test67 {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		try {
			List<String> pokemonList = new ArrayList<>();
			HashMap<String, Integer> pokemonMap = new HashMap<>();
			StringTokenizer st = new StringTokenizer(br.readLine());
			int n = Integer.parseInt(st.nextToken());
			int m = Integer.parseInt(st.nextToken());
			for (int i = 1; i <= n; i++) {
				String name = br.readLine();
				pokemonList.add(name);
				pokemonMap.put(name, i);
			}
			for (int i = 0; i < m; i++) {
				String lookFor = br.readLine();
				if (lookFor.matches("\\d+")) {
					int key = Integer.parseInt(lookFor);
					bw.write(pokemonList.get(key - 1));
				} else {
					bw.write(String.valueOf(pokemonMap.get(lookFor)));
				}
				if (i < m - 1) {
					bw.write("\n");
				}
			}
			bw.flush();
			bw.close();
			br.close();
		} catch (NumberFormatException | IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

}