프로그래밍/알고리즘 풀이
[node.js] 트리의 부모 찾기 ( 백준 11725번 )
카카수(kakasoo)
2021. 4. 15. 13:56
반응형
const readline = require("readline");
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout,
});
let N = 0;
const input = [];
rl.on("line", (line) => {
if (!N) {
N = Number(line);
} else {
input.push(line);
if (input.length === N - 1) {
main();
}
}
});
const main = () => {
const graph = [];
const answer = [];
for (let i = 1; i <= N; i++) {
graph[i] = [];
}
input.forEach((edge) => {
const [from, to] = edge.split(" ");
graph[from].push(to);
graph[to].push(from);
});
const bfs = (start) => {
const visited = new Array(N);
visited[start] = true;
const queue = [start];
while (queue.length) {
const cur = queue.shift();
for (let i = 0; i < graph[cur].length; i++) {
const next = graph[cur][i];
if (!visited[next]) {
visited[next] = true;
answer[next] = cur;
queue.push(next);
}
}
}
};
bfs(1);
// console.time();
// answer.forEach((ans) => console.log(ans));
let result = "";
answer.forEach((ans) => (result += ans + "\n"));
console.log(result);
// console.timeEnd();
};
문제 자체는 쉬우니깐 마지막 부분을 봐줬으면 한다. 아마도 이 문제를 어렵다고 생각하지는 않을테니, 다들 시간초과 때문에 더 좋은 풀이가 있나 찾아 다녔을 것이다.
결과부터 말하자면 forEach로 결과를 하나씩 출력하면 ( 또는 forEach가 아니어도 어쨌든 console.log를 n-1번 써야 한다면 ),
주석처리가 되지 않은 부분처럼 문자열을 다 합쳐서 한 번의 console.log를 찍는 것보다 느리다.
어느 정도로 느리냐면 단순한 예제에서 8배 느렸다.
물린 하나는 2 밀리세컨드고 하나는 17 밀리세컨드 정도니깐 8배라는 말은 어불성설일 수 있으나,
저 주석처리한 부분과 주석처리 하지 않은 부분 하나만 가지고도 문제를 통과하고 통과하지 못하고가 갈렸다.
이 정도면 지금까지 시간초과로 실패한 문제들을 전부 고칠 수 있을 것 같다.
이 부분에서만큼은 자바스크립트 만세를 외칠 수가 없다...
반응형