[Baekjoon] 1406 에디터(파이썬/자바스크립트/NodeJS)

이번 문제는 아래 링크에서 풀어볼 수 있습니다.

1406번: 에디터

---

문제

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

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

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

LDBP $

커서를 왼쪽으로 한 칸 옮김 (커서가 문장의 맨 앞이면 무시됨)

커서를 오른쪽으로 한 칸 옮김 (커서가 문장의 맨 뒤이면 무시됨)

커서 왼쪽에 있는 문자를 삭제함 (커서가 문장의 맨 앞이면 무시됨) 삭제로 인해 커서는 한 칸 왼쪽으로 이동한 것처럼 나타나지만, 실제로 커서의 오른쪽에 있던 문자는 그대로임

$라는 문자를 커서 왼쪽에 추가함

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

입력

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

출력

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

예제 입력 1

abcd
3
P x
L
P y

예제 출력 1

abcdyx

예제 입력 2

abc
9
L
L
L
L
L
P x
L
B
P y

예제 출력 2

yxabc

예제 입력 3

dmih
11
B
B
P x
L
B
B
B
P y
D
D
P z

예제 출력 3

yxz

 

---

정답 코드

나의 답안(시간 초과)

import sys


def mySolution():
    string = list(sys.stdin.readline().rstrip())
    idx = len(string)

    for _ in range(int(sys.stdin.readline().rstrip())):
        data = sys.stdin.readline().rstrip().split()
        oper = data[0]

        if oper == "P":
            string.insert(idx, data[1])
            idx += 1
        elif oper == "L" and idx > 0:
            idx -= 1
        elif oper == "D" and idx < len(string):
            idx += 1
        elif oper == "B" and idx > 0:
            del string[idx - 1]
            idx -= 1

    return "".join(string)


print("mySolution", mySolution())

모범 답안
파이썬 Ver.

import sys


def answer1():
    l_stack = list(sys.stdin.readline().rstrip())
    r_stack = []

    for _ in range(int(sys.stdin.readline().rstrip())):
        data = sys.stdin.readline().rstrip().split()
        oper = data[0]

        if oper == "P":
            l_stack.append(data[1])
        elif oper == "L" and l_stack:
            r_stack.append(l_stack.pop())
        elif oper == "D" and r_stack:
            l_stack.append(r_stack.pop())
        elif oper == "B" and l_stack:
            l_stack.pop()

    return "".join(l_stack + list(reversed(r_stack)))


print("answer1", answer1())

자바스크립트/NodeJS Ver.

const fs = require('fs');
const filePath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
let input = fs.readFileSync(filePath).toString().trim().split('\n');

const str = input
    .splice(0, 1)[0]
    .split('')
    .filter((val) => val !== '\r');

const n = parseInt(input.splice(0, 1));

const answer1 = (str, data) => {
    let lStack = str;
    let rStack = [];

    for (let d of data) {
        const oper = d[0];

        if (oper === 'P') {
            lStack.push(d[2]);
        } else if (oper === 'L' && lStack.length) {
            rStack.push(lStack.pop());
        } else if (oper === 'D' && rStack.length) {
            lStack.push(rStack.pop());
        } else if (oper === 'B' && lStack.length) {
            lStack.pop();
        }
    }

    return lStack.concat(rStack.reverse()).join('');
};

console.log(answer1(str, input));

 

---

문제 풀이

 나는 우선 입력 받은 문자열을 리스트로 변환해놓고 문자열 맨 끝에 커서가 놓인 다는 점에서 착안하여 문자열의 길이만큼을 인덱스로 설정한 뒤 이 인덱스와 문자열 리스트를 조건에 따라 변화를 주는 식으로 풀었다. 허나 insert 메서드의 시간 복잡도가 O(n)이라서 시간 초과가 떴다. 구글링을 해보니 시간 복잡도가 O(1)인 pop과 append 메서드를 활용하여 문제를 풀어야 했다.

 우선 별도의 스택 r_stack을 선언한다. 그 다음 조건이 'L'이고 l_stack에 요소가 들어 있다면 l_stack을 pop하여 r_stack에 append한다. 즉 커서는 l_stack의 끝에 계속 머물러 있는 셈이고 r_stack에는 l_stack에서 pop한 요소들이 순서대로 쌓인다. 이로써 커서를 왼쪽으로 한 칸 옮기는 기능을 구현할 수 있다. 조건이 'D'이면서 r_stack에 요소가 있다면 r_stack에서 요소를 pop하여 l_stack에 append한다. 이러면 l_stack의 길이가 한 칸 늘어나면서 커서는 오른쪽으로 한 칸 이동한 셈이 된다. 

 즉 명시적으로 커서의 인덱스를 선언해서 조작하는 것이 아니라 스택 두 개의 요소를 주고 받으면서 추가 및 삭제, 그리고 커서의 좌우 이동을 구현하는 것이다.

---

함께 보기

https://my-coding-notes.tistory.com/229

[🥈3 / 백준 1406 / 파이썬] 에디터

https://seongonion.tistory.com/53

[백준] 1406번 에디터 - 파이썬(Python)

https://kbwplace.tistory.com/87

[Python/파이썬] 백준 알고리즘 1406번 - 에디터