백준 1406번: 에디터 (Python) TODO

문제

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

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

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

LDBP $

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

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

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

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

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

입력

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

출력

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

 

 

---

 

처음 풀이(틀렸음 안 봐도 됨)

리스트에 문자열 넣고 경우 나눠서 지우고, 자르고, 넣고... 하는 방법으로 코드를 짰는데, 시간 초과가 나온다.

cf) 커서를 오른쪽으로 한 칸 움직이는 걸 R로 입력을 받아서(L이 있으니 R이 있겠지..하는 생각?) 한 30분동안 고민했다. 문제를 잘 읽자..

 

코드

import sys

string = sys.stdin.readline().strip()
cursor = len(string)
N = int(sys.stdin.readline())
for i in range(N):
    command = sys.stdin.readline().split()
    if command[0] == "L":
        if cursor > 0:
            cursor -= 1

    if command[0] == "D":
        if cursor < len(string):
            cursor += 1

    if command[0] == "B":
        if cursor != 0:
            if len(string) == cursor:
                string = string[:cursor-1]
                cursor -= 1
            else:
                string = string[:cursor-1] + string[cursor:]
                cursor -= 1

    if command[0] == "P":
        if len(string) == cursor:
            string = string + command[1]
            cursor += 1
        else:
            string = string[:cursor] + command[1] + string[cursor:]
            cursor += 1

print(string)

 

---

 

모범적인 풀이

 

백준 내 아이디 'bupjae'님 말에 따르면,

• python 의 list 는 연속된 메모리 공간을 할당해서 자료를 저장합니다.
• list의 한가운데에 원소를 삽입하거나, 또는 한가운데 있는 원소를 삭제하기 위해서는 그 자리 이후의 원소들을 모두 한 칸씩 밀거나 당겨야 합니다.
• 이 문제를 제대로 풀기 위해서는 다음 중 한 가지를 선택해야 합니다.

1. list 의 맨 뒤에서만 삽입/삭제 연산을 할 수 있도록 알고리즘을 구현하기
2. 한가운데의 원소를 삽입하거나 삭제했을 때 바로 앞뒤의 원소 이외의 원소를 건드릴 필요가 없는 자료구조를 사용하기

라고 한다. 1번 풀이를 하자. 2번 풀이는 TODO.

 

풀이 1

https://bnzn2426.tistory.com/112 의 풀이를 완전 따라한 것. 자세한 설명은 이 분 블로그에서 써있다.

블로그에 풀이 뿐만 아니라 time complexity 등의 고민한 내용이 많이 적혀있어, 보고 배우기 좋다.

리스트로 스택 2개를 만들어서, 커서 기준으로 왼쪽에 있는 원소는 스택 1에, 오른쪽에 있는 원소는 스택 2에 저장한다는 게 핵심 아이디어다.

 

코드

import sys

stack1 = list(sys.stdin.readline().rstrip())
stack2 = []
M = int(sys.stdin.readline())

for i in range(M):
    command = sys.stdin.readline().rstrip()

    if command[0] == "L":
        if stack1 != []:
            stack2.append(stack1.pop())

    if command[0] == "D":
        if stack2 != []:
            stack1.append(stack2.pop())

    if command[0] == "B":
        if stack1 != []:
            stack1.pop()

    if command[0] == "P":
        stack1.append(command[2])

answer = stack1 + stack2[::-1]
print(''.join(answer))