kakasoo

[node.js] 트리의 부모 찾기 ( 백준 11725번 ) 본문

프로그래밍/알고리즘 풀이

[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배라는 말은 어불성설일 수 있으나,
저 주석처리한 부분과 주석처리 하지 않은 부분 하나만 가지고도 문제를 통과하고 통과하지 못하고가 갈렸다.

이 정도면 지금까지 시간초과로 실패한 문제들을 전부 고칠 수 있을 것 같다.
이 부분에서만큼은 자바스크립트 만세를 외칠 수가 없다...

반응형