🎲백준 11725번 트리의 부모 찾기

백준 11725번

💊 시간초과가 발생했었는데 탐색한 노드를 체크할 때, checked 배열에다 노드 값을 push하고 includes로 확인했다. 이 부분을 아래 코드와 같이 수정하니 맞았다.

코드

const fs = require('fs'); 
const input = fs.readFileSync('/dev/stdin').toString().trim().split('\n');
const N = +input.shift();

const tree = Array.from(new Array(N + 1), () => []);

for (let i = 0; i < N - 1; i++) {
  const [a, b] = input[i].split(" ").map(Number);
  tree[a].push(b);
  tree[b].push(a);
}

const parents = [];

let willCheck = [1];
let checked = new Array(N+1).fill(0);

while (willCheck.length) {
  const check = willCheck.pop();
  checked[check] = 1;
  for (let i = 0; i < tree[check].length; i++) {
    const child = tree[check][i];
    if (checked[child] === 0) {
      parents[child-2] = check;
      willCheck.push(child);
    }
  }
}


console.log(parents.join("\n"));