[백준] 1406

에디터

문제

한 줄로 된 간단한 에디터를 구현하려고 한다. 이 편집기는 영어 소문자만을 기록할 수 있는 편집기로, 최대 600,000글자까지 입력할 수 있다.

이 편집기에는 '커서'라는 것이 있는데, 커서는 문장의 맨 앞(첫 번째 문자의 왼쪽), 문장의 맨 뒤(마지막 문자의 오른쪽), 또는 문장 중간 임의의 곳(모든 연속된 두 문자 사이)에 위치할 수 있다. 즉 길이가 L인 문자열이 현재 편집기에 입력되어 있으면, 커서가 위치할 수 있는 곳은 L+1가지 경우가 있다.

이 편집기가 지원하는 명령어는 다음과 같다.

초기에 편집기에 입력되어 있는 문자열이 주어지고, 그 이후 입력한 명령어가 차례로 주어졌을 때, 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 구하는 프로그램을 작성하시오. 단, 명령어가 수행되기 전에 커서는 문장의 맨 뒤에 위치하고 있다고 한다.

입력

첫째 줄에는 초기에 편집기에 입력되어 있는 문자열이 주어진다. 이 문자열은 길이가 N이고, 영어 소문자로만 이루어져 있으며, 길이는 100,000을 넘지 않는다. 둘째 줄에는 입력할 명령어의 개수를 나타내는 정수 M(1 ≤ M ≤ 500,000)이 주어진다. 셋째 줄부터 M개의 줄에 걸쳐 입력할 명령어가 순서대로 주어진다. 명령어는 위의 네 가지 중 하나의 형태로만 주어진다.

출력

첫째 줄에 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 출력한다.

풀이

const editor = () => {
  const fs = require("fs");
  let input = fs.readFileSync("/dev/stdin").toString().trim().split("\\n");
  let index = 2;
  const len = input.length;
  let stackLeft = [];
  let stackRight = [];

  for (let i in input[0]) {
    stackLeft.push(input[0][i]);
  }

  while (index < len) {
    switch (input[index][0]) {
      case "L":
        if (stackLeft.length !== 0) {
          stackRight.push(stackLeft.pop());
        }
        break;
      case "D":
        if (stackRight.length !== 0) {
          stackLeft.push(stackRight.pop());
        }
        break;
      case "B":
        if (stackLeft.length !== 0) {
          stackLeft.pop();
        }
        break;
      case "P":
        stackLeft.push(input[index][2]);
        break;
    }
    index++;
  }
  console.log([...stackLeft, ...stackRight.reverse()].join(""));
};

editor();

반성회

현재 커서보다 오른쪽에 있는 값은 stackRight에 넣고, 왼쪽에 있는 값은 stackLeft에 넣는다stackLeft 스택의 top 값은 커서와 가장 가까운 값 (stackLeft 안의 값들 중 가장 오른쪽에 있는 값)

stackRight 스택의 top 값은 커서와 가장 가까운 값 (stackRight 안의 값들 중 가장 왼쪽에 있는 값)

에디터에는 초기에 입력받은 문자열이 들어있으며, 커서는 문자열의 맨 오른쪽에 있다

따라서 stackLeft는 입력받은 문자열로, stackRight는 빈 스택으로 초기화한다

 

스택 사용 X (메모리 초과)

const editor = () => {
  const fs = require("fs");
  let input = fs.readFileSync("/dev/stdin").toString().trim().split("\n");
  let list = new LinkedList();
  let str = [...input[0]];
  let cursor = str.length + 1;

  //  h e l l o
  //  0 1 2 3 4
  // 0 1 2 3 4 5 -> 커서의 왼쪽: 인덱스 -1

  for (let i in input) {
    if (parseInt(i) === 0 || parseInt(i) === 1) continue;
    switch (input[i][0]) {
      case "L":
        if (cursor > 0) cursor--;
        break;
      case "D":
        if (cursor < str.length + 1) cursor++;
        break;
      case "B":
        if (cursor > 0) {
          str = [...str.slice(0, cursor), ...str.slice(cursor + 1)];
          cursor--;
        }
        break;
      case "P":
        str = [...str.slice(0, cursor), input[i][2], ...str.slice(cursor)];
        break;
    }
  }
  console.log(str.join(""));
};

editor();

연결리스트 구현 (시간 초과)

class Node {
  constructor(data, next = null, prev = null) {
    this.data = data;
    this.next = next;
    this.prev = prev;
  }
}

class LinkedList {
  constructor() {
    this.head = new Node(-1);
    this.tail = new Node(-1);
    this.head.next = this.tail;
    this.tail.prev = this.head;
    this.size = 0;
  }
  insertLast(data) {
    let node = new Node(data);
    let currentNode;

    currentNode = this.tail.prev;
    node.prev = currentNode;
    node.next = this.tail;
    this.tail.prev = node;
    currentNode.next = node;
    this.size++;
  }
  insertLeft(data, node) {
    let newNode = new Node(data);
    newNode.next = node;
    newNode.prev = node.prev;
    node.prev.next = newNode;
    node.prev = newNode;
    this.size++;
  }
  deleteLeft(node) {
    if (node === this.head.next) return;
    node.prev = node.prev.prev;
    node.prev.next = node;
    this.size--;
  }
  printAll() {
    let currentNode = this.head.next;

    if (currentNode) process.stdout.write(currentNode.data);
    while (currentNode.next !== this.tail) {
      currentNode = currentNode.next;
      process.stdout.write(currentNode.data);
    }
  }
}

const editor = () => {
  const fs = require("fs");
  let input = fs.readFileSync("/dev/stdin").toString().trim().split("\n");
  let list = new LinkedList();
  const len = input.length;
  let index = 2;
  let node;

  for (let i in input[0]) {
    list.insertLast(input[0][i]);
  }
  node = list.tail;

  while (index < len) {
    switch (input[index][0]) {
      case "L":
        if (node !== list.head.next) {
          node = node.prev;
        }
        break;
      case "D":
        if (node !== list.tail) {
          node = node.next;
        }
        break;
      case "B":
        if (node !== list.head.next) {
          list.deleteLeft(node);
        }
        break;
      case "P":
        list.insertLeft(input[index][2], node);
        break;
    }
    index++;
  }
  list.printAll();
};

editor();

연결리스트로 풀면 시간이 무조건 초과난다

(연결리스트 초기화하는 시간, 연결리스트 next와 prev 옮기는 시간 등 + 시간 제한이 타이트함)

따라서 커서 기준으로 왼쪽은 stackLeft, 오른쪽은 stackRight에 넣어서 push, pop을 통한 O(1) 풀이법이 정석