🎲 백준 15681번 트리와 쿼리

백준 15681번

🎨 참고 코드
💊 출력을 따로 하면 시간초과가 발생하므로 배열에 저장한 뒤 join을 사용해 한 번에 출력한다.

코드

const fs = require('fs'); 
const input = fs.readFileSync('/dev/stdin').toString().trim().split('\n');
const [N, R, Q] = input.shift().split(" ").map(Number);

const tree = [];

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

const result = new Array(N + 1).fill(0);
const visited = new Array(N + 1).fill(0);

const dfs = (node) => {
  let count = 1;
  for (let i = 0; i < tree[node].length; i++) {
    const next = tree[node][i];
    if (visited[next]) {
      count += result[next];
    } else {
      visited[next] = 1;
      count += dfs(next);
    }
  }
  result[node] = count;
  return count;
};

visited[R] = 1;
dfs(R);

const answer = [];
for (let i = 0; i < Q; i++) {
  answer.push(result[Number(input[i + N - 1])]);
}
console.log(answer.join("\n"));