신입 개발자에서 시니어 개발자가 되기까지

[백준 nodeJS] 18258 큐2 본문

javascript/알고리즘

[백준 nodeJS] 18258 큐2

Jin.K 2022. 10. 19. 23:10

시간복잡도가 O(1)이어야 한다길래 따로 큐를 만드는 풀이방법을 썼다. 아마 코드의 길이로만 따진다면 가장 길었던 것 같다.

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 여섯 가지이다.

push X: 정수 X를 큐에 넣는 연산이다.
pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 큐에 들어있는 정수의 개수를 출력한다.
empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

const [n, ...commands] = require("fs")
  .readFileSync("example.txt")
  .toString()
  .trim()
  .split("\n");

class Queue {
  constructor() {
    this.first = null;
    this.last = null;
    this.size = 0;
  }
  /*
  로직 
  1.맨 뒤에 추가해서, 맨 앞에서 제거하는 방법
  2. 맨 앞에 추가해서, 맨 뒤에서 제거하는 방법 */
  enqueue(val) {
    let newNode = new Node(val);
    if (this.size === 0) {
      this.first = newNode;
      this.last = newNode;
    } else {
      this.last.next = newNode;
      this.last = newNode;
    }
    return ++this.size;
  }
  dequeue() {
    if (this.size === 0) return null;
    let dequeuedNode = this.first;
    if (this.size === 1) {
      this.last = null;
    }
    this.first = this.first.next;
    dequeuedNode.next = null;
    this.size--;
    return dequeuedNode;
  }
}

class Node {
  constructor(value) {
    this.value = value;
    this.next = null;
  }
}

function dealWithCommands(loopCondition, commandArray) {
  let queue = new Queue();
  let result = [];
  loopCondition = +loopCondition;
  for (let i = 0; i < loopCondition; i++) {
    if (commandArray[i].split(" ")[0] === "push") {
      queue.enqueue(commandArray[i].split(" ")[1]);
    }
    switch (commandArray[i]) {
      case "pop":
        queue.size > 0 ? result.push(queue.dequeue().value) : result.push(-1);
        break;
      case "size":
        result.push(queue.size);
        break;
      case "empty":
        queue.size > 0 ? result.push(0) : result.push(1);
        break;
      case "front":
        queue.size > 0 ? result.push(queue.first.value) : result.push(-1);
        break;
      case "back":
        queue.size > 0 ? result.push(queue.last.value) : result.push(-1);
        break;
      default:
        break;
    }
  }
  return result;
}
const result = dealWithCommands(n, commands);
console.log(result.join("\n"));