[python] 백준 1406번 에디터 문제 풀이

https://www.acmicpc.net/problem/1406

1406번: 에디터

 

 

문제

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

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

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

 

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

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

입력

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

출력

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

 

 

 

 

 

---

 

 

 

💻 코드

 

import sys

input = sys.stdin.readline

str1 = list(input().rstrip())
str2 = list()
m = int(input())

for _ in range(m):
    command = list(input().split())

    if command[0] == 'L':
        if str1:
            str2.append(str1.pop())

    elif command[0] == 'D':
        if str2:
            str1.append(str2.pop())

    elif command[0] == 'B':
        if str1:
            str1.pop()

    else:
        str1.append(command[1])

print(''.join(str1 + list(reversed(str2))))

 

 

 

 

 

📝 풀이

 

이 문제는 처음에 봤을 때 단순하게 입력 받은 명령어가 L이면 cursor를 -1하고, D이면 cursor를 +1하고, B이면, 문자열에서 인덱스 cursor-1의 문자를 pop()이나 removed()를 이용하여 삭제하고, 그 외(P)이면 $라는 문자를 문자열에서 인덱스 cursor에 위치하도록 insert() 함수를 사용하면 된다고 생각했다. 하지만 이와 같이 코드를 작성하게 되면 예상했듯이 시간 초과가 나온다. 특히 insert() 함수가 O(N)의 시간복잡도를 갖기 때문에 시간이 매우 오래 걸린다.

 

 

시간 초과가 나오는 코드

 

import sys

input = sys.stdin.readline

sentence = list(input().rstrip())
m = int(input())
cursor = len(sentence)

for _ in range(m):
    command = list(input().split())

    if command[0] == 'L':
        if cursor > 0:
            cursor -= 1

    elif command[0] == 'D':
        if cursor < len(sentence):
            cursor += 1

    elif command[0] == 'B':
        if cursor > 0:
            cursor -= 1
            sentence.remove(sentence[cursor])

    else:
        sentence.insert(cursor, command[1])
        cursor += 1

print(''.join(sentence))

 

 

 

해결 방법

 

시간을 줄이기 위해 insert() 함수 대신 push하는 함수 append()를 사용하면 이 함수는 시간복잡도가 O(1)이기 때문에 확실히 줄일 수 있다. append()를 사용하면서 원하는 위치에 삽입하려면 cursor를 인덱스로 사용하는 것이 아니라 cursor를 기준으로 리스트를 2개를 사용하여 문자를 이동시킴으로써 커서 위치가 이동하도록 만들어야 한다.

 

 

커서를 왼쪽으로 이동시킬 때는 str1에서 마지막 문자를 pop하여 str2에 push 하도록 하고, 커서를 오른쪽으로 이동시킬 때는 str2에서 마지막 문자를 pop하여 str1에 push 하도록 한다. 하지만 마지막에 리스트 str1과 str2를 합쳐 출력하기 전에 str2 리스트에 있는 문자를 reversed 함수를 통해 순서를 뒤바꿔주어야 하는데, 그 이유는 예를 들어 명령어가 L인 경우, str1에서 pop하여 pop한 문자를 str2 리스트의 맨 앞에 삽입이 되어야 하는데 append 함수를 사용해야하기 때문에 맨 뒤에 삽입이 되기 때문이다.

 

 

print(''.join(str1 + list(reversed(str2))))

 

출력 코드는 다음과 같으며 reversed를 사용하는 대신에 list에서 제공하는 reverse 함수를 사용해도 상관 없다.

 

 

---

 

※ 참고

 

https://wikidocs.net/14

02-3 리스트 자료형

 

 

파이썬의 주요 함수 시간 복잡도

https://www.ics.uci.edu/~pattis/ICS-33/lectures/complexitypython.txt

 

 

https://docs.python.org/ko/3/library/functions.html?highlight=reversed#reversed 

내장 함수 — Python 3.10.5 문서

 

 

https://itholic.github.io/python-reverse-reversed/

[python] reverse, reversed 차이