백준 1920번 수 찾기 (자바)
출처
1920번: 수 찾기
첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들
www.acmicpc.net
문제
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보다 작다.
출력
- M개의 줄에 답을 출력한다. 존재하면 1을, 존재하지 않으면 0을 출력한다.
입출력 예

접근 방법
처음 포스팅하는 이분탐색 문제다.
기초적인 이분탐색 문제로, 처음과 끝을 기준으로 중간값을 구한 후 target값을 찾을 때까지 범위를 줄여가며 중간값이 target 값과 같으면 반환한다.
1. Arrays.sort()를 사용해 배열을 정렬시킨다. (이분 탐색은 정렬이 돼있어야 가능한 알고리즘이기 때문)
2. mid번쨰 원소가 target보다 큰 경우 왼쪽 부분을 탐색하기 위해 end = mid-1이다.
3. mid번째 원소가 target보다 같거나 큰 경우 오른쪽 부분을 탐색하기 위해 start = mid+1이다.
4. mid번째 원소가 target과 같은 경우 mid를 반환한다.
5. start가 end보다 큰 경우는 해당 원소를 찾지 못한 것이므로 -1을 반환한다.
내 코드
package binarySearch; | |
import java.io.BufferedReader; | |
import java.io.IOException; | |
import java.io.InputStreamReader; | |
import java.util.Arrays; | |
import java.util.StringTokenizer; | |
public class FindNumber { | |
/** | |
* 백준 1920 수 찾기 (https://www.acmicpc.net/problem/1920) | |
*/ | |
public static void main(String[] args) throws NumberFormatException, IOException { | |
//정수 N개 | |
//X 존재하는지 알아내기 | |
BufferedReader reader = new BufferedReader(new InputStreamReader(System.in)); | |
int n = Integer.parseInt(reader.readLine()); | |
int[] array = new int[n]; | |
StringTokenizer st = new StringTokenizer(reader.readLine()); | |
for (int i=0; i<n; i++) { | |
array[i] = Integer.parseInt(st.nextToken()); | |
} | |
Arrays.sort(array); | |
int m = Integer.parseInt(reader.readLine()); | |
int[] result = new int[m]; | |
st = new StringTokenizer(reader.readLine()); | |
for (int i=0; i<m; i++) { | |
int num = Integer.parseInt(st.nextToken()); | |
int temp = binarySearch(array, num, 0, n-1); | |
result[i] = temp != -1 ? 1 : 0; | |
} | |
for (int i=0; i<m; i++) { | |
System.out.println(result[i]); | |
} | |
} | |
private static int binarySearch(int[] array, int target, int start, int end) { | |
if (start > end) { | |
//시작이 커지면 일치하는것이 없다. | |
return -1; | |
} | |
int mid = (start+end)/2; | |
if (array[mid] == target) { | |
//같을 때는 해당 위치가 구하고자 하는 위치 | |
return mid; | |
} else if (array[mid] > target) { | |
//배열의 mid값이 목표값보다 크면 end를 줄인다. | |
return binarySearch(array, target, start, mid-1); | |
} else { | |
//배열의 mid값이 목표값보다 작으면 start를 늘린다. | |
return binarySearch(array, target, mid+1, end); | |
} | |
}//binarySearch | |
} | |
고려할 점
1. 이진탐색의 개념을 파악할 것
2. start와 end를 활용하여 mid를 구할 것
3. 어떤 경우에 어떤 수를 반환해야 하는지 파악할 것
'알고리즘 > 백준' 카테고리의 다른 글
[백준 14888] 연산자 끼워넣기 (자바) (0) | 2020.12.29 |
---|---|
[백준 1715] 카드 정렬하기 (자바) (2) | 2020.12.29 |
[백준 14500] 테트로미노 (자바) (0) | 2020.12.23 |
[백준 11048] 이동하기 (자바) (0) | 2020.12.23 |
[백준 14499] 주사위 굴리기 (자바) (0) | 2020.12.23 |