[이코테] 이진탐색 - 부품 찾기 - JAVA

최영환·2022년 10월 18일

이코테

목록 보기

14/24

💡 문제

동빈이네 전자 매장에는 부품이 N개 있다. 각 부품은 정수 형태의 고유한 번호가 있다. 어느 날 손님이 M개의 종류의 부품을 대량으로 구매하겠다며 당일 날 견적서를 요청했다. 동빈이는 때를 놓치지 않고 손님이 문의한 부품 M개 종류를 모두 확인해서 견적서를 작성해야 한다. 이때 가게 안에 부품이 모두 있는지 확인하는 프로그램을 작성해보자.
예를 들어 가게의 부품이 총 5개일 때 부품 번호가 다음과 같다고 하자.

N=5
[8, 3, 7, 9, 2]

손님은 총 3개의 부품이 있는지 확인 요청했는데 부품 번호는 다음과 같다.

M=3
[5, 7, 9]

이때 손님이 요청한 부품 번호의 순서대로 부품을 확인해 부품이 있으면 yes를, 없으면 no를 출력한다. 구분은 공백으로 한다.

입력

첫째 줄에 정수 N이 주어진다. (1<=N<=1,000,000)
둘째 줄에는 공백으로 구분하여 N개의 정수가 주어진다. 이때 정수는 1보다 크고 1,000,000 이하이다.
셋째 줄에는 정수 M이 주어진다. (1<=M<=100,000)
넷째 줄에는 공백으로 구분하여 M개의 정수가 주어진다. 이때 정수는 1보다 크고 10억 이하이다.

출력

첫째 줄에 공백으로 구분하여 각 부품이 존재하면 yes를, 없으면 no를 출력한다.

💬 입출력 예시

입력

출력

no yes yes

📌 풀이(소스코드)

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

// 부품 찾기
public class BinarySearch_01 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        // N 과 N 개의 부품 입력
        int n = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        int[] items = new int[n];
        for (int i = 0; i < n; i++) {
            items[i] = Integer.parseInt(st.nextToken());
        }
        // items 정렬
        Arrays.sort(items);

        // M 입력
        int m = Integer.parseInt(br.readLine());
        // M 개의 부품 입력과 동시에 이진 탐색 수행
        st = new StringTokenizer(br.readLine(), " ");
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < m; i++) {
            int target = Integer.parseInt(st.nextToken());
            // 찾으면 "yes" 찾지 못하면 "no" 를 StringBuilder 에 추가
            if (binarySearch(items, target, 0, n-1) == -1) {
                sb.append("no").append(" ");
            }
            else {
                sb.append("yes").append(" ");
            }
        }
        System.out.println(sb);
    }
    
    // 이진 탐색 메소드
    public static int binarySearch(int[] arr, int target, int start, int end) {
        while (start <= end) {
            int mid = (int)((start + end) / 2);
            // 중간점의 값이 찾고자 하는 값과 같을 경우 중간점의 인덱스 반환
            if (arr[mid] == target) {
                return mid;
            }
            // 좌측 탐색
            else if (arr[mid] > target) {
                end = mid - 1;
            }
            // 우측 탐색
            else {
                start = mid + 1;
            }
        }
        // 못찾을 경우 -1 반환
        return -1;
    }
}