[Python] 백준 5397 : 키로거

스택과 큐 문제 중 하나인 키로거 문제를 풀어보았습니다. 난이도는 중정도로 될 것 같습니다.

 

 

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

5397번: 키로거

 

풀이 방식은 스택을 두 개로 사용하는 방법을 선택하였습니다. 처음에는 시뮬레이션 그대로 코딩을 짜주다가 시간 제한이 뜨게 되어 스택을 사용하는 방법을 찾아보았더니 간단한 해답이 나왔습니다.

왼쪽 스택과 오른쪽 스택으로 나눠주어 그 사이에 커서를 두고 진행하면 간단합니다. 단어를 넣거나 뺄 때는 왼쪽에서 append와 pop을 이용하고, 커서를 왼쪽으로 움직일 때는 왼쪽 스택을 pop해주고 그 값을 오른쪽 스택에 push를 해주는 방법을 선택하였습니다. 커서를 오른쪽으로 움직이는 경우는 반대로 진행하였습니다.

마지막 출력 부분은 오른쪽 스택을 반대로 반전시켜 왼쪽 스택 뒤에 붙여주어 한 단어로 출력해주게 코드를 작성하였습니다.

 

t = int(input())

for _ in range(t):
    left_stack = []
    right_stack = []
    data = input()
    for i in data:
        if i == '-':
            if left_stack:
                left_stack.pop()
        elif i == '<':
            if left_stack:
                right_stack.append(left_stack.pop())
        elif i == '>':
            if right_stack:
                left_stack.append(right_stack.pop())
        else:
            left_stack.append(i)
    left_stack.extend(reversed(right_stack))
    print(''.join(left_stack))

 

조건문 중간에 스택이 비어있을 조건을 주지 않아서 애먹긴 했지만 디버깅을 진행하여, 조건문을 다시 수정하여 정답을 도출하였습니다.

 

오늘도 감사합니다.