[백준] 1920 수 찾기 Node.js (Binary Search)

Janet·2023년 8월 3일
1

Algorithm

목록 보기
252/314

문제

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'));
profile
😸

0개의 댓글