python/알고리즘 문제풀이

BOJ 5397. 키로거 (Python)

륵기 2020. 5. 5. 15:03
728x90
반응형

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

 

5397번: 키로거

문제 창영이는 강산이의 비밀번호를 훔치기 위해서 강산이가 사용하는 컴퓨터에 키로거를 설치했다. 며칠을 기다린 끝에 창영이는 강산이가 비밀번호 창에 입력하는 글자를 얻어냈다. 키로거는 사용자가 키보드를 누른 명령을 모두 기록한다. 따라서, 강산이가 비밀번호를 입력할 때, 화살표나 백스페이스를 입력해도 정확한 비밀번호를 알아낼 수 있다. 강산이가 비밀번호 창에서 입력한 키가 주어졌을 때, 강산이의 비밀번호를 알아내는 프로그램을 작성하시오. 입력 첫째 줄에 테

www.acmicpc.net

'<' 와 '>' 로인한 커서 구분, 그리고 '-' 일때 원소 삭제인 것을 유의하고 문제를 풀자.

스택을 이용하면 쉽게 풀리며, 2개의 스택을 이용해보자.

 

키보드로 입력된 문자들은 왼쪽 스택에 계속 입력된다고 가정하고,

'<'가 입력됬을 때 왼쪽 스택의 원소를 오른쪽에 보낸다.

'>'가 입력됬을 때는 오른쪽 스택의 원소를 왼쪽으로 보내고

'-' 가 입력됬을 때는 pop하면 된다.

 

테스트 케이스의 예제를 통해서 설명해보도록 하겠다.

<<BP>A 까지 입력됬을 때,  BP가 입력되고, P는 오른쪽 스택에 보내지게된다.

그러면 최종적으로 나중에 출력할 때는 BAP 순으로 나올수 있게 된다. 물론 이는 오른쪽 스택의 값을 최종적으로

왼쪽 스택에 다시 넣어주어야한다. 

 

왼쪽 스택

B A

 

오른쪽 스택

P  
t = int(input())
for i in range(0,t):
    s1 = []
    s2 = []
    li = input()
    for j in range(len(li)):
        if li[j] == '<':
            if s1:
                s2.append(s1.pop())

        elif li[j] == '>':
            if s2:
                s1.append(s2.pop())

        elif li[j] == '-':
            if s1:
                s1.pop()
        else:
            s1.append(li[j])
    s1.extend(reversed(s2))
    print(''.join(s1))
728x90
반응형