Algorithm/이론
이분 탐색 (Binary Search)
goakgoak
2020. 4. 24. 16:14
이분 탐색 (Binary Search)
이분 탐색 알고리즘의 큰 골자를 이루는 로직은
1. 오름차순으로 정렬 되어 있는 자료 구조에서 특정값을 찾을 때,
2. 탐색 범위를 절반씩 나눠가면서 해당 값을 찾아가는 것이다.
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개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다. 모든 정수의 범위는 -231 보다 크거나 같고 231보다 작다.
www.acmicpc.net
위 문제를 이분 탐색으로 풀어보았다.
배열의 중간값(mid)을 기준으로
target 값 보다 작으면 binarySearch(start ~ mid-1)
target 값 보다 크면 binarySearch(mid+1 ~ end) 를 호출한다.
소스코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
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[] a;
static boolean[] c;
static int n, m;
static StringTokenizer st;
static StringBuilder sb;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
n = stoi(br.readLine());
a = new int[n];
st = new StringTokenizer(br.readLine());
for (int i = 0; i < n; i++) {
a[i] = stoi(st.nextToken());
}
Arrays.sort(a);
m = stoi(br.readLine());
st = new StringTokenizer(br.readLine());
for (int i = 0; i < m; i++) {
System.out.println(binarySearch(0, n - 1, stoi(st.nextToken())));
}
}
static int binarySearch(int start, int end, int target) {
if (start > end)
return 0;
int mid = (start + end) / 2;
if (target == a[mid])
return 1;
if (target < a[mid]) {
return binarySearch(start, mid - 1, target);
} else {
return binarySearch(mid + 1, end, target);
}
}
static int stoi(String s) {
return Integer.valueOf(s);
}
}
|
cs |