[python] 백준 5397 키로거

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

5397번: 키로거

 

문제해결

• 포인터를 기준으로 왼쪽 오른쪽에 스택을 두고 이동시키는 것을 생각했다.
• 왼쪽으로 움직이면 왼쪽 스택의 마지막 것을 오른쪽에 넣는다.
• 오른쪽으로 움직이면 오른쪽 스택 마지막 것을 왼쪽에 넣는다.
• 지우면 왼쪽 스택의 마지막것을 지우면 된다.
• 사실 시간이 적기 때문에 시간초과가 나서 고민했다. 마지막 완성된 두 스택을 연결해서 정답을 추출할 때 ''.join(iterable)로 했다. (이게 for문 보다 조금 더 빠르다. (하지만 N이 커지면 O(N) 으로 봐도 무방)

 

CODE

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    L = []
    R = []
    string = str(input().rstrip())
    for word in string:
        if word not in ['-', '<', '>']:
            L.append(word)
            continue
        elif word == '>':
            if len(R):
                L.append(R.pop())
        elif len(L):
            if word == '-':
                L.pop()
            elif word == '<':
                R.append(L.pop())
    ans = ''.join(L) + ''.join(R[::-1])
    # for i in L:
    #     ans = ans + i
    # while len(R):
    #     ans = ans + R.pop()
    print(ans)