코딩연습이 좋아서/이론이 좋아서

Sorting and Searching(정렬, 이분검색과 결정알고리즘) - 이분검색

zoaseo 2024. 12. 11. 17:34

설명

임의의 N개의 숫자가 입력으로 주어집니다. N개의 수를 오름차순으로 정렬한 다음 N개의 수 중 한 개의 수인 M이 주어지면

이분검색으로 M이 정렬된 상태에서 몇 번째에 있는지 구하는 프로그램을 작성하세요. 단 중복값은 존재하지 않습니다.

입력

첫 줄에 한 줄에 자연수 N(3<=N<=1,000,000)과 M이 주어집니다.

두 번째 줄에 N개의 수가 공백을 사이에 두고 주어집니다.

출력

첫 줄에 정렬 후 M의 값의 위치 번호를 출력한다.

예시 입력 1 

8 32
23 87 65 12 57 32 99 81

예시 출력 1

3

 

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 m = sc.nextInt();
        int[] a = new int[n];
        for (int i = 0; i < n; i++) {
            a[i] = sc.nextInt();
        }
        Arrays.sort(a);
        int answer = 0;
        int lt = 0;
        int rt = n - 1;
        while (lt <= rt) {
            int mid = (lt + rt) / 2;
            if (a[mid] == m) {
                answer = mid + 1;
                break;
            } else if (a[mid] > m) rt = mid - 1;
            else lt = mid + 1;
        }

        System.out.println(answer);
    }
}

- lt와 rt를 기본으로 세팅하고 mid = (lt + rt) / 2인 mid 값과 비교하여 한쪽을 날리고 난 뒤 (lt<=rt) 조건까지 반복하여 비교한다.