[알고리즘] 키로거 5397번

 

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

5397번: 키로거

이 문제는 스택을 활용하여 푸는 문제이다.

 

이 문제의 아이디어는 스택을 2개를 만들어서 커서를 스택 2개의 중간지점이라고 생각하면 된다.

1. 데이터를 삽입하면 왼쪽 스택에 데이터를 넣는다.
2. '-'가 나오면 왼쪽 스택에 마지막 원소를 뺀다.
3. '<'가 나오면 커서가 왼쪽으로 이동하는 것이므로, 왼쪽 스택에서 오른쪽 스택으로 원소를 옮긴다.
4. '>'가 나오면 반대로 오른쪽 스택에서 왼쪽 스택으로 원소를 옮기면 된다.

이 방식으로 코드를 구현하면 아래처럼 구현된다.

n=int(input())

for i in range(n):
  pwd = str(input()) #비밀번호입력데이터
  left_stack = []
  right_stack = []
  for j in pwd:
    if j == '<':
      if left_stack:
        right_stack.append(left_stack.pop())
    elif j == '>':
      if right_stack:
        left_stack.append(right_stack.pop())
    elif j == '-':
      if left_stack:
        left_stack.pop()
    else:
      left_stack.append(j)
  left_stack.extend(reversed(right_stack)) #두개 스택 합치기
  print(''.join(left_stack))