알고리즘/백준

[알고리즘] 키로거 5397번

레아킴 2022. 8. 15. 19:25
반응형

 

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

 

5397번: 키로거

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한줄로 이루어져 있고, 강산이가 입력한 순서대로 길이가 L인 문자열이 주어진다. (1 ≤ L ≤ 1,000,000) 강산이가 백스페이스를 입

www.acmicpc.net

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

 

이 문제의 아이디어는 스택을 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))
반응형