BOJ_5397 [ 키로거 ]

 

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

5397번: 키로거

 

문제 요구 사항

주어진 문자에서 요구 사항에 따라 동작 후 비밀번호를 출력하면 되는 문제이다.

요구 사항은 총 3가지가 있다.

1. "<" : 커서가 왼쪽으로 이동한다.

2. ">" : 커서가 오른쪽으로 이동한다.

3. "-" : 커서 앞에 글자를 지운다.

 

스택을 활용하여 문제를 해결할 수 있다. 

스택을 사용하는 이유는 커서 이동 후 문자를 중간에 삽입하거나 삭제시 뒤에 문자들을 그대로 복원하기 위해서 LIFO 특성인 스택을 사용한다.

 

 # answer , stk 리스트 선언 후 
 "<"  : 커서를 왼쪽으로 이동하므로 answer에 맨 마지막 문자를 stk에 푸시한다.
 ">"  : 커서를 오른쪽으로 이동하므로 stk에 맨 마지막 문자를 answer에 푸시한다.
 "-"  : 커서 앞 문자를 제거해야 되므로 answer에 맨 마지막 문자를 제거한다.

 

※ 정답 출력하기 위해 for문을 사용하게 되면 시간 초과가 발생하니 for문 사용안하고 해결해야 된다.

testCase = int(input())
for t in range(testCase):
    tmp = list(input())
    answer = []
    stk = []
    for i in range(len(tmp)):
        if tmp[i] == "<":
            if len(answer) > 0:
               stk.append(answer.pop())
        elif tmp[i] == ">":
            if len(stk) > 0:
                answer.append(stk.pop())
        elif tmp[i] == "-":
            if len(answer) > 0:
                answer.pop()
        else:
            answer.append(tmp[i])
    # 시간초과 해결 코드
    answer.extend(reversed(stk))
    print(''.join(answer))

    '''
    시간 초과 코드
    for i in range(len(answer)):
        result += answer[i]
    while len(stk) > 0:
        result += stk.pop()
    print(result)
    '''