문제
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을 출력한다.
예제 입력 1
5
4 1 5 2 3
5
1 3 7 9 5예제 출력 1
1
1
0
0
1문제풀이
💡 문제풀이 과정
- N의 최대값 100,000, M의 최대값도 100,000이다. 단순히 순차적으로 탐색하여 비교 연산을 할 경우, 최악의 경우 100,000 x 100,000 = 10,000,000,000번(100억번)의 비교 연산이 필요하다. 시간 제한은 1초로 최대 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'));