[백준][JAVA]1620번 : 나는야 포켓몬 마스터 이다솜

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

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

 

---

사실 단순하게, 문자열로 입력받으면 해당 문자열의 index 반환

숫자(index)로 입력을 받으면 해당 index의 문자열을 반환한다.

 

처음엔 그냥 ArrayList에서 검색하는 방식을 했더니, 시간초과가 떴다.

그래서 map과 배열을 이용해서 푸는 코드로 바꾸었다.

 

코드는, 시간초과로 실패했던 코드와

성공한 코드 모두 첨부한다.

 

 

 

1. ArrayList로하여 시간초과가 났던 코드

import java.util.*;

class Main {
    public static void main(String args[]) throws Exception {
        Scanner kb = new Scanner(System.in);
        int n=kb.nextInt();
        int m=kb.nextInt();

        ArrayList<String> list=new ArrayList<>();
        for(int i=0; i<n; i++) list.add(kb.next());

        for(int i=0; i<m; i++){
           String temp=kb.next();
           if(list.contains(temp)){
               System.out.println((list.indexOf(temp))+1);
           }else{
               System.out.println((list.get(Integer.parseInt(temp)-1)));
           }
        }

    }
}

 

 

2. Map + 배열로 성공한 코드

import java.util.*;

class Main {
    public static void main(String args[]) throws Exception {
        Scanner kb = new Scanner(System.in);
        int n=kb.nextInt();
        int m=kb.nextInt();

        HashMap<String, Integer> map = new HashMap<>();
        String[] arr=new String[n];
        for(int i=0; i<n; i++){
            String s=kb.next();
            arr[i]=s;
            map.put(s,i+1);
        }
        for(int i=0; i<m; i++){
            if(kb.hasNextInt()) System.out.println(arr[kb.nextInt()-1]);
            else System.out.println(map.get(kb.next()));
        }
    }
}