백준 1620 나는야 포켓몬 마스터 이다솜(Java)

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

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

---

 

해석 및 팁

 

---

 

이 문제는 2개의 HashMap을 사용하여 풀 수 있습니다. 하나는 포켓몬의 이름을 key로 저장하고 다른 하나는 번호를 key로 저장합니다. 그 후에 반복문을 통해 숫자를 입력받으면 이름을 출력하고 입력을 입력받으면 번호를 출력하도록 하면 됩니다.

 

---

 

Java 코드

 

---

import java.util.Scanner;
import java.util.HashMap;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    StringBuilder sb = new StringBuilder();
    HashMap<String, Integer> name = new HashMap<>();
    HashMap<Integer, String> num = new HashMap<>();
    

    int n = sc.nextInt();
    int m = sc.nextInt();

    for(int i = 1; i <= n; i++) {
      String str = sc.next();
      name.put(str, i);
      num.put(i, str);
    }

    for(int i = 0; i < m; i++) {
      String str = sc.next();
      if(Character.isDigit(str.charAt(0))) sb.append(num.get(Integer.parseInt(str))+"\n");
      else sb.append(name.get(str)+"\n");
    }

    System.out.println(sb);
  }
}