[Java] BOJ 1920 수 찾기

문제 

문제 링크 https://www.acmicpc.net/problem/1920

• N개의 정수 A[1], A[2], …, A[N]이 주어져 있을 때, 이 안에 X라는 정수가 존재하는지 알아내는 프로그램을 작성하시오.
• 첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다.
• 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다.
• 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다.
• 다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다.
• 모든 정수의 범위는 -231 보다 크거나 같고 231보다 작다.
• M개의 줄에 답을 출력한다. 존재하면 1을, 존재하지 않으면 0을 출력한다.

아이디어

• hashmap을 사용하여 key값과 value값으로 값을 저장해주는데 입력 들어오는 값을 key값으로 넣어주고 value값을 1로 넣어준다.
• 그리고 찾아야되는 수를 입력받으면서 getOrDefault를 사용해서 해당 수가 hashmap에 key값으로 저장되어 있으면 value값인 1을 출력하고 없으면 default로 저장되어 있는 0을 출력하도록 했다. 

겪은 시행착오

• X

getOrDefault를 처음 써 봤는데, 확실히 기억해뒀다가 나중에도 잘 써야겠다..!

코드

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

public class BOJ1920 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        int N = Integer.parseInt(br.readLine());

        // A값들을 저장할 hashmap
        // key 값으로 value를 찾는 형식으로, 검색에 유용하기 때문
        HashMap<Integer, Integer> hash = new HashMap<Integer, Integer>();

        StringTokenizer st = new StringTokenizer(br.readLine());

        // 입력받은 값들을 hashmap에 key값과 value 값인 1을 저장
        for(int i = 0; i < N; i++){
            int n = Integer.parseInt(st.nextToken());

            hash.put(n, 1);
        }

        int M = Integer.parseInt(br.readLine());

        StringTokenizer st2 = new StringTokenizer(br.readLine());

        for(int i = 0; i < M; i++){
            // 찾을 값
            int n2 = Integer.parseInt(st2.nextToken());

            // getOrDefault는 hashmap에서 key값이 없을 때, null값을 출력하지 않고
            // default로 정해진 값을 출력함
            // hashmap에서 찾을 값이 key값으로 있다면 해당 value 값인 1을 저장
            // 없다면 0을 저장
            sb.append(hash.getOrDefault(n2, 0) + "\n");
        }

        System.out.println(sb);
    }
}