[JAVA] 백준 10815번 숫자 카드

문제

 

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.

 

입력

 

첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.

셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다

 

출력

 

첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 가지고 있으면 1을, 아니면 0을 공백으로 구분해 출력한다.

 

입력예제

 

5
6 3 2 10 -10
8
10 9 -5 2 3 4 5 -10

 

출력예제

 

1 0 0 1 1 0 0 1

 

풀이

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        // 첫번째 숫자 hashmap으로 입력받기 키에 '6 3 2 10 -10' 넣기
        int n = Integer.parseInt(bf.readLine());
        HashMap<Integer, Integer> arrn = new HashMap<>();
        for (int i = 0; i < 1; i++) {
            String[] x = bf.readLine().split(" ");
            for (int j = 0; j < n; j++) {
                arrn.put(Integer.parseInt(x[j]), j);
            }
        }


        // 두번째 숫자 배열로 입력받기
        int m = Integer.parseInt(bf.readLine());
        int[] arrm = new int[m];

        // 배열 안에 입력 받기
        for (int i = 0; i < 1; i++) {
            // split 로 잘라서 arr 배열 안에 넣기
                String[] y = bf.readLine().split(" ");
            for (int j = 0; j < m; j++) {
                arrm[j] = Integer.parseInt(y[j]);
            }
        }


        // arrn 에 arrm[i] 키가 포함되어있는지 확인하고
        for (int i = 0; i < m; i++) {
            int a = 0;
            if (arrn.containsKey(arrm[i])) {
                // 있으면 a에 +1
                a += 1;
            }else {
                // 없으면 a에 +0
                a += 0;
            }
            // 결과를 출력
            bw.write(a + " ");
        }

        bw.flush();
        bw.close();
    }
}

 

 

 

 

10815번: 숫자 카드