yunicornlab

백준 18352번 특정 거리의 도시 찾기 JavaScript 풀이 [BFS] 본문

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'))
반응형