Coding Test/Practice
백준 18352번 특정 거리의 도시 찾기 JavaScript 풀이 [BFS]
yunicornlab
2024. 7. 26. 20:36
반응형
백준 18352번 특정 거리의 도시 찾기 문제를 자바스크립트로 BFS 알고리즘을 이용해서 풀어보았다.
https://www.acmicpc.net/problem/18352
let fs = require("fs");
let input = fs.readFileSync('/dev/stdin').toString().trim().split("\n");
class Queue {
constructor() {
this.items = {};
this.head = 0;
this.tail = 0;
}
enqueue(element) {
this.items[this.tail] = element;
this.tail++;
}
dequeue() {
const element = this.items[this.head];
delete this.items[this.head];
this.head++;
return element;
}
getLength() {
return this.tail - this.head;
}
}
// 도시 개수 N, 도로 개수 M, 거리 K, 출발 도시 번호 X
let [n, m, k, x] = input[0].split(' ').map(Number);
let graph = new Array(n+1).fill().map(_ => []);
for (let i=1; i<=m; i++) {
let [start, end] = input[i].split(' ').map(Number);
graph[start].push(end);
}
let distance = new Array(n+1).fill(-1);
queue = new Queue();
queue.enqueue(x);
// x에서 x로 가는 거리는 항상 0
distance[x] = 0;
while (queue.getLength() != 0) {
let cur = queue.dequeue();
for (let next of graph[cur]) {
if (distance[next] == -1) {
queue.enqueue(next);
distance[next] = distance[cur] + 1;
}
}
}
answer = [];
distance.map((v, i) => {
if (v == k) answer.push(i);
})
answer.length == 0 ? console.log(-1) : console.log(answer.join('\n'))
반응형