백준 1920번: 수 찾기 [JAVA]

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

1920번: 수 찾기

• 풀이

이진 탐색을 이용하여 배열에서 수를 찾는 함수를 만든다.

숫자를 받을 때마다 수 찾기 함수를 실행시켜 존재 여부를 출력한다.

• 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
	static int N, M;
	static int[] A;
	
	static int search(int num) {
		int high = N-1;
		int low = 0;
		while (low<=high) {
			int mid = (high+low) / 2;
			
			if (A[mid]==num) return 1; // 찾음
			else if (A[mid]>num) high = mid - 1; // 중앙값이 찾을 숫자보다 큰 경우
			else if (A[mid]<num) low =mid + 1; // 중앙값이 찾을 숫자보다 작은 경우
		}
		return 0;
	}

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(bf.readLine());
		
		// 입력
		N = Integer.parseInt(st.nextToken());
		A = new int[N];
		st = new StringTokenizer(bf.readLine());
		for (int i=0; i<N; i++) A[i] = Integer.parseInt(st.nextToken());
		
		// 정렬
		Arrays.sort(A);
		
		M = Integer.parseInt(bf.readLine());
		st = new StringTokenizer(bf.readLine());
		for (int i=0; i<M; i++) {
			int num = Integer.parseInt(st.nextToken()); // 찾을 숫자 입력
			System.out.println(search(num)); // 수 찾고 출력
		}
	}
}