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
4 1 5 2 3
5
1 3 7 9 5
1
1
0
0
1
💡 문제풀이 과정
이진탐색
과 Set()
을 이용한 풀이를 생각했다. 이진탐색의 경우, M개의 수를 순회(N) x logN으로 시간복잡도는 O(NlogN)
이다. 즉, 100,000 x log₂100,000 = 100,000 x 17 (반올림) = 약 170만번이므로 가능하다.Set()
으로 답안 제출하니 처음에 시간초과가 발생했었는데, console.log()
를 2번이상 사용했더니 그러했다. console.log()
는 수행시간이 길다고하니 될 수 있으면 1번만 할 것.✅ 답안 #1: 이진탐색, while문을 이용한 풀이 (메모리: 39840KB, 시간: 376ms)
const filePath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
const input = require('fs').readFileSync(filePath).toString().trim().split('\n');
const endIdx = Number(input[0]) - 1;
const A = input[1].split(' ').map(Number).sort((a, b) => a - b);
const B = input[3].split(' ').map(Number);
function binarySearch(arr, target, start, end) {
while (start <= end) {
const mid = Math.floor((start + end) / 2);
if (target === arr[mid]) return 1;
else if (target > arr[mid]) {
start = mid + 1;
} else {
end = mid - 1;
}
}
return 0;
}
const answer = B.map((b) => binarySearch(A, b, 0, endIdx));
console.log(answer.join('\n'));
✅ 답안 #2: Set()을 이용한 풀이 (메모리: 44096KB, 시간: 296ms)
const filePath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
const input = require('fs').readFileSync(filePath).toString().trim().split('\n');
const A = input[1].split(' ').map(Number);
const B = input[3].split(' ').map(Number);
const aArr = new Set(A);
const answer = B.map((b) => (aArr.has(b) ? 1 : 0));
console.log(answer.join('\n'));