[백준저지] 1920번 : 수 찾기

www.acmicpc.net/problem/1920

1. 문제

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안에 존재하는지 알아내면 된다. 모든 정수의 범위는 -2^31 보다 크거나 같고 2^31보다 작다.

출)

M개의 줄에 답을 출력한다. 존재하면 1을, 존재하지 않으면 0을 출력한다.

2. 풀이

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] a = new int[n];
        for (int i = 0; i < a.length; i++) {
            a[i] = sc.nextInt();
        }
        int m = sc.nextInt();
        int[] b = new int[m];
        for (int i = 0; i < b.length; i++) {
            b[i] = sc.nextInt();
        }
        // 우선 a를 정렬해놓고
        Arrays.sort(a);
        // b를 loop 돌면서 a에 존재하는지 확인
        for (int i = 0; i < b.length; i++) {
            int result = 0;
            int start = 0;
            int end = a.length-1;
            // 이진 탐색하면서
            while (start <= end) {
                int mid = (start + end) / 2;
                // 찾으면 1로
                if (a[mid] == b[i]) {
                    result = 1;
                    break;
                } else if (a[mid] > b[i]) {
                    end = mid - 1;
                } else if (a[mid] < b[i]) {
                    start = mid + 1;
                }
            }
            System.out.println(result);
        }
    }
}

이진 탐색을 이용해서 해결.