incastle의 콩나물

[알고리즘] 백준 5397번, 키로거 , 스택 , python(22) 본문

python/알고리즘

[알고리즘] 백준 5397번, 키로거 , 스택 , python(22)

incastle 2020. 4. 5. 20:14

문제

창영이는 강산이의 비밀번호를 훔치기 위해서 강산이가 사용하는 컴퓨터에 키로거를 설치했다. 며칠을 기다린 끝에 창영이는 강산이가 비밀번호 창에 입력하는 글자를 얻어냈다.

키로거는 사용자가 키보드를 누른 명령을 모두 기록한다. 따라서, 강산이가 비밀번호를 입력할 때, 화살표나 백스페이스를 입력해도 정확한 비밀번호를 알아낼 수 있다.

강산이가 비밀번호 창에서 입력한 키가 주어졌을 때, 강산이의 비밀번호를 알아내는 프로그램을 작성하시오.

 

 

아이디어

- stack을 두 개 준비한다. 

- 커서가 움직임에 따라서 스택에서 pop과 append를 반복한다.

 

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))
Comments