본문 바로가기
코딩테스트

[이진 탐색] 백준 1920번 수 찾기 자바(Java)

by Enhydra lutris 2023. 1. 10.

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

 

1920번: 수 찾기

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들

www.acmicpc.net

문제

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을 출력한다.

코드

import java.util.*;

public class No1920 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[] arr = new int[N];
        for (int i = 0; i < N; i++) {
            arr[i] = sc.nextInt();
        }
        Arrays.sort(arr); // 이진 탐색을 하기 위해서는 정렬이 되어있어야 한다.
        int M = sc.nextInt();


        for (int i = 0; i < M; i++) {
            if (binarySearch(arr, sc.nextInt()))
                System.out.println(1);
            else
                System.out.println(0);

        }
    }


    public static boolean binarySearch(int[] arr, int key) {
        int left = 0;
        int right = arr.length - 1;

        while (left <= right) {
            int mid = (left + right) / 2;    // 중간위치

            if (key < arr[mid]) right = mid - 1;  // key값이 중간보다 작을때
            else if (key > arr[mid]) left = mid + 1; // key값이 중간보다 클때
            else return true;  // key값과 중간값이 같을때
        }
        return false; // 값이 존재 하지 않을때
    }
}

풀이

이진 탐색을 이용하여 입력받은 값이 존재하는지 확인후 출력

이진 탐색 코드&방법

아래에 있는 링크에 설명되어 있음

https://seaotter.tistory.com/21

 

[이진 탐색] 이진탐색 방법 & 자바 코드

이진탐색 이진 탐색은 정렬된 리스트에서 검색 범위를 절반씩 줄여가며 탐색하는 알고리즘이다. 시간복잡도: O(log n) 이진탐색 하는 방법 1. 이진탐색할 배열을 정렬한다. 2. 처음에는 배열의 0번

seaotter.tistory.com

 

댓글