[BOJ] 1406 에디터

www.acmicpc.net/problem/1406

1406번: 에디터

풀이

• 처음에는 case로 나누고 조건에 맞게 문자열 추가하는 방법 수행(실패, 시간초과)

  # 실패
  edit_text = str(input())
  edit_text_len = len(edit_text)
  
  cur_pointer = edit_text_len
  command_num = int(input())
  
  for _ in range(command_num):
      command = input().split(' ')
      if command[0] == 'L' and cur_pointer > 0:
          cur_pointer -= 1
      elif command[0] == 'D' and cur_pointer < edit_text_len:
          cur_pointer += 1
      elif command[0] == 'B' and cur_pointer > 0:
          edit_text = edit_text[:cur_pointer-1] + edit_text[cur_pointer:]
          edit_text_len -= 1
          cur_pointer -= 1
      elif command[0] == 'P':
          edit_text = edit_text[:cur_pointer] +command[1] +edit_text[cur_pointer:]
          cur_pointer += 1
          edit_text_len += 1
      else:
          assert("Not L, D, B, P")
  print(edit_text)

   

• list를 사용한 stack 2개를 사용해 문자열 삽입, 삭제 시 O(1)로 수행.

  from sys import stdin
  
  # 가운데 커서가 있고, left_stack은 오른쪽이 입구, right_stack은 입구가 왼쪽에 있다고 생각
  left_stack = list(input())
  right_stack = []
  
  command_num = int(input())
  for command in stdin:
      # 커서를 왼쪽으로 옮김.
      if command[0] == 'L' and left_stack:
          # 왼쪽 스택의 끝 값 -> 오른쪽 스택에 추가.
          right_stack.append(left_stack.pop())
          
      # 커서를 오른쪽으로 옮김
      elif command[0] == 'D' and right_stack:
          left_stack.append(right_stack.pop())
          
      # 커서 왼쪽 문자열 삭제
      elif command[0] == 'B' and left_stack:
          left_stack.pop()
          
      # 커서 왼쪽에 문자열 추가
      elif command[0] == 'P':
          left_stack.append(command[2])
  
  # right_stack은 거꾸로 출력
  print(''.join(left_stack + right_stack[::-1]))