[Ch.06 - SortSearch] 08. 이분검색 ##

8. 이분검색

 

설명

임의의 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 int solution(int n, int m, int[] arr) {
		int answer = 0;
		Arrays.sort(arr);
		int lt = 0;
		int rt = arr.length;
		int mid = 0;
		Arrays.sort(arr);

		while (mid != m) {
			mid = (lt + rt) / 2;
			if (arr[mid] == m)
				return answer = mid+1;
			
			else {
				if (arr[mid] > m) {
					rt = mid - 1;
				} else if (arr[mid] < m) {
					lt = mid + 1;
				}
			}
		}

		return answer;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner kb = new Scanner(System.in);
		int n = kb.nextInt();
		int m = kb.nextInt();
		int[] arr = new int[n];
		for (int i = 0; i < n; i++) {
			arr[i] = kb.nextInt();
		}
		System.out.println(T.solution(n, m, arr));
	}

}

1. 정렬

2. 중간 지점 위치의 값과 해당 값 비교

3. 해당 위치의 값이 더 크면 lt를 mid-1로,

4. 해당 위치의 값이 더 작으면 rt를 mid+1로 설정

5. 같을 때 까지 반복

 

+) 세련된 코드

package sortsearch.ch06;

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

public class Sortsearch08 {
	// # 이분 검색
	// 임의의 N개의 숫자 입력으로 주어지면, N개의 수를 오름차순 정렬 후
	// N개 중 한 개의 수 M으로, M이 정렬된 상태에서 이분검색으로 몇 번째에 있는지 구함
	// 중복값은 존재하지 않는다.

	public static void main(String[] args) {
		Sortsearch08 T = new Sortsearch08();
		Scanner kb = new Scanner(System.in);
		int n = kb.nextInt();
		int m = kb.nextInt();
		int[] arr = new int[n];

		for (int i = 0; i < n; i++) {
			arr[i] = kb.nextInt();
		}
		System.out.println(T.solution(n, m, arr));
		kb.close();
	}

	public int solution(int n, int m, int[] arr) {
		int answer = 0;
		// 결과 변수 0으로 가정
		Arrays.sort(arr);
		// 배열 정렬

		int lt = 0, rt = n - 1;
		// 왼쪽 포인터, 오른쪽 포인터 [0부터이므로 n-1]
		while (lt <= rt) {
			// 왼쪽포인터가 오른쪽포인터보다 작거나 같을때 반복
			int mid = (lt + rt) / 2;
			// 왼쪽포인터와 오른쪽포인터를 이용해 중간점 설정

			if (arr[mid] == m) {
				answer = mid + 1;
				break;
				// 주어진 수와 중간점의 값이 같으면
				// 결과변수에 중간점위치+1을 설정 [0부터 n-1번째로 세었기 때문에]
			}
			if (arr[mid] > m)
				rt = mid - 1;
			// 주어진 수보다 중간점의 값이 크면
			// 오른쪽포인터를 중간점-1로 설정 [중간점의 왼쪽으로 한칸 땡김]

			else
				lt = mid + 1;
			// 주어진 수보다 중간점의 값이 작으면
			// 왼쪽 포인터를 중간점+1로 설정 [중간점의 오른쪽으로 한칸 땡김]

		}

		return answer;
	}

}

---

import java.util.*;
  
public class Main {
  public static void main(String[] args){
    Scanner in=new Scanner(System.in);
    int n=in.nextInt();
    int k=in.nextInt();
    int[] arr = new int[n];
    for(int i=0;i<n;i++){
      arr[i]=in.nextInt();
    }
    Arrays.sort(arr);
    //for(int i=0;i<n;i++) if(arr[i]==k) System.out.println(i+1); 
    int lt=0;
    int rt=n;
    int mid=0;
    while(mid!=k){
      mid=(lt+rt)/2;
      if(arr[mid]==k){
        System.out.println(mid+1);
        break;
      }
      else {
        if(arr[mid]<k){
          lt=mid+1;
        }
        else{
          rt=mid-1;
        }
      }
    }
  }
}