문제 링크

성능 요약

메모리: 48184 KB, 시간: 616 ms

분류

이분 탐색, 자료 구조, 해시를 사용한 집합과 맵, 정렬

제출 일자

2025년 1월 31일 02:51:36

문제 설명

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을 출력한다.

풀이

느낀점

• 이분탐색 맛보기로 괜찮았다.
• 조금씩 난이도를 높여서 다양한 이분탐색 기준을 구현해봐야겠다.

설계 : 5분

• 입력된 숫자들을 오름차순으로 배열에 정렬하고, binarySearch 메서드를 구현했다.
• 처음 left, right 인덱스 변수를 초기화하고, 정확히 같은수가 존재하는지 확인해야 하기 때문에 left ≤ right인 동안 이분탐색을 진행한다.
• 주어진 수보다 nums[mid]가 작으면 right를 mid-1로, 주어진 수가 더 작으면 left를 mid+1로 인덱스를 수정한다.
• 정확히 같은수를 찾으면 true를 리턴하고, while문이 끝날때까지 찾지 못하면 false를 리턴한다.

코드(Java)

• 구현 시간: 20분

/**
 * Author: yngbao97, Yuk Yejin
 * Problem: 수 찾기_1920
 * Date: 2025.01.31
 */

import java.util.*;
import java.lang.*;
import java.io.*;

public class Main {
	static BufferedReader br;
	static BufferedWriter bw;
	static StringTokenizer st;
	static int[] nums;

	public static void main(String[] args) throws Exception {

		br = new BufferedReader(new InputStreamReader(System.in));
		bw = new BufferedWriter(new OutputStreamWriter(System.out));
		
		int n = Integer.parseInt(br.readLine());
		nums = new int[n];
		st = new StringTokenizer(br.readLine(), " ");
		for (int i = 0; i < n; i++) nums[i] = Integer.parseInt(st.nextToken());
		Arrays.sort(nums);

		int m = Integer.parseInt(br.readLine());
		st = new StringTokenizer(br.readLine(), " ");
		for (int i = 0; i < m; i++) {
			int num = Integer.parseInt(st.nextToken());
			if (binarySearch(num)) bw.write("1\n");
			else bw.write("0\n");
		}
		bw.flush();
		bw.close();
		br.close();
	}

	private static boolean binarySearch(int num) {
		int left = 0;
		int right = nums.length - 1;
		
		while (left <= right) {
			int mid = (left + right) / 2;
			if (nums[mid] < num) left = mid + 1;
			else if (nums[mid] > num) right = mid - 1;
			else return true;
		}
		return false;
	}
}