[백준 1406 : 에디터] 파이썬 풀이

1406번: 에디터

---

문제

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

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

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

 

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

 

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

입력

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

출력

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

 

---

✏️ 풀이

 

left, right로 stack을 두 개 구현하여 풀이하였다.

left의 가장 위를 커서라고 생각하고, 커서가 이동하는 것이 아니라 left의 맨 위 요소 또는 right의 맨 아래 요소가 움직이는 것으로 구현하였다. 

• P x: x라는 문자를 추가할 때는, left stack에 추가
• L : 커서를 왼쪽으로 움직이므로 left stack에서 right stack으로 하나의 요소 이동
• D: 커서를 오른쪽으로 움직이므로 right stack에서 left stack으로 하나의 요소 이동
• B: 커서 왼쪽의 문자를 삭제하므로 left stack의 하나를 제거 

 

🚫 이때 right stack에서는 append로 값을 넣어서 뒤에 값이 추가되었으므로 마지막에 reverse를 해주었다.

 

---

💻 코드

import sys
input = sys.stdin.readline

left = list(input().rstrip())
n = int(input())
right = []

for _ in range(n):
    a = str(input()).rstrip() 
    if a == "L":
        if left: #left가 비지 않았다면 
            right.append(left.pop()) #left에서 하나를 빼서 right에 넣어줌 
    elif a == "D": 
        if right:#right가 비지 않았다면
            left.append(right.pop()) #right에서 하나를 빼서 left에 넣어줌
    elif a == "B":
        if left: #left가 비지 않았다면
            left.pop() #left에서 하나를 제거
    else:
        P, x = a.split() 
        left.append(x) #x라는 값을 left에 넣어줌 

# left와 right를 결합하여 결과 문자열을 생성
result = ''.join(left) + ''.join(right[::-1])  #right는 거꾸로 바꿔서 더해줌
print(result) #결과 출력

---

⚠️ 잘못된 풀이 _ 시간초과

처음에 문제 보고 생각나는데로 푼 풀이 

-> cursor 변수를 만들어서 현재 커서의 위치(인덱스)로 활용하는 방법

• 입력값이 L일 때는, cursor 값을 1 감소
• 입력값이 D일 때는 cursor 값을 1 증가
• B일 때는 cursor 인덱스에 위치한 문자를 삭제하고 cursor 값을 1 감소
• P x 일 때는 cursor+1 위치(커서 왼쪽)에 x값을 insert s.insert(cursor+1, num)하고 cursor 값을 1 증가

 

import sys
input = sys.stdin.readline

s = list(input().rstrip())
n = int(input()) 
cursor = len(s)-1

for _ in range(n):
  a = str(input()).rstrip()
  if a == "L":
    if cursor >= 0:
      cursor = cursor - 1
  elif a == "D":
    if cursor != len(s)-1:
      cursor = cursor + 1
  elif a == "B":
    if cursor >= 0:
      del s[cursor]
      cursor = cursor - 1
  else: 
    a, num = a.split()
    s.insert(cursor+1, num)
    cursor = cursor +1
    
print(*s, sep="")

 

 

insert와 del은 시간복잡도가 O(N)이라 시간초과가 날 가능성이 많아 사용할 때 주의해야 한다고 한다.

 

 

---

📍자료형 & 메서드 별 시간복잡도 참고 링크

Python 자료형별 시간 복잡도 정리