https://www.acmicpc.net/problem/1920
[문제]
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보다 작다.
[코드]
import java.util.*;
public class BaekJoon_1920 {
public static int n;
public static int[] num;
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc=new Scanner(System.in);
n=sc.nextInt();
num=new int[n];
for(int i=0;i<n;i++)
num[i]=sc.nextInt();
Arrays.sort(num); // 이진 탐색할 배열은 반드시 정렬되어 있어야 한다.
int m=sc.nextInt();
for(int i=0;i<m;i++) {
int find=sc.nextInt();
System.out.println(binarysearch(num,find,0,n-1));
}
}
// 이진 탐색
public static int binarysearch(int[] arr,int target, int start, int end) {
while(start<=end) {
int mid=(start+end)/2;
if(arr[mid]==target) return 1;
else if(arr[mid]>target) end=mid-1;
else start=mid+1;
}
return 0;
}
}
[고찰]
이번 문제는 탐색하고자 하는 수를 이진탐색을 사용하여 해결하는 것이다. 이진탐색 코드는 재귀 함수를 사용하여 구현하는 방법과 while 문을 사용하여 구현하는 방법 두 가지가 있는데 모두 암기하고 있는 것이 좋다.
이진탐색을 사용할 때 가장 중요한 점은 반드시 정렬된 배열을 사용해야 한다는 점이다. 이 부분만 주의한다면 쉽게 해결할 수 있는 문제였다.
'백준' 카테고리의 다른 글
[정수론 및 조합론_11051] 이항 계수 2 (0) | 2021.07.09 |
---|---|
[백준_2740번] 행렬 곱셈 (0) | 2021.07.09 |
[백준_1629번] 곱셈 (0) | 2021.07.08 |
[백준_1780번] 종이의 개수 (0) | 2021.07.08 |
[백준_1992번] 쿼드트리 (0) | 2021.07.08 |