[BOJ] #5397_키로거(파이썬, Python3)

본 게시글은 백준 저지 알고리즘 문제를 '파이썬, Python3' 언어로 풀이한 내용을 주관적으로 정리하였으며, 내용과 관련된 코드리뷰 및 피드백 환영합니다.


저번에 풀었던 커서 관련된 문제인 '에디터' 문제와 굉장히 유사하다.
'에디터' 문제를 풀고나면 이 문제는 식은 죽 먹기로 풀 수 있으니 '에디터' 문제를 풀어보는 것을 권한다.

[ 에디터 문제 풀이 ]
https://gracekim-devstory.tistory.com/24

---

티어

Silver II

문제

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

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

강산이가 비밀번호 창에서 입력한 키가 주어졌을 때, 강산이의 비밀번호를 알아내는 프로그램을 작성하시오. 
강산이는 키보드로 입력한 키는 알파벳 대문자, 소문자, 숫자, 백스페이스, 화살표이다.

입력

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한줄로 이루어져 있고, 강산이가 입력한 순서대로 길이가 
L인 문자열이 주어진다. (1 ≤ L ≤ 1,000,000) 강산이가 백스페이스를 입력했다면, '-'가 주어진다. 
이때 커서의 바로 앞에 글자가 존재한다면, 그 글자를 지운다. 화살표의 입력은 '<'와 '>'로 주어진다. 
이때는 커서의 위치를 움직일 수 있다면, 왼쪽 또는 오른쪽으로 1만큼 움직인다. 나머지 문자는 비밀번호의 일부이다. 
물론, 나중에 백스페이스를 통해서 지울 수는 있다. 만약 커서의 위치가 줄의 마지막이 아니라면, 
커서 및 커서 오른쪽에 있는 모든 문자는 오른쪽으로 한 칸 이동한다.

출력

각 테스트 케이스에 대해서, 강산이의 비밀번호를 출력한다. 비밀번호의 길이는 항상 0보다 크다.

예제 입력 및 출력 1

2
<<BP<A>>Cd-
ThIsIsS3Cr3t

BAPC
ThIsIsS3Cr3t

---

풀이

import sys
input = sys.stdin.readline

n = int(input())

for _ in range(n):
    l = input().rstrip()
    word_l = []
    word_r = []

    for i in l:
        if i == '<':
            if word_l:
                word_r.append(word_l.pop())

        elif i == '>':
            if word_r:
                word_l.append(word_r.pop())

        elif i == '-':
            if word_l:
                word_l.pop()

        else:
            word_l.append(i)
    print("".join(word_l + list(reversed(word_r))))

이 문제의 가장 관건은 커서의 위치이다. 커서를 기준으로 예를 들어, 'ABCD_'라는 글자를 쓰고, 'AB_CD'로 커서를 옮겼다고 치자. (_가 대충 커서로 생각하면 될 듯 하다. 커서를 기준으로 백스페이스를 눌렀을 때 바로 앞 글자가 사라져야 하는 기능을 구현해야 한다. 그러기 위해서는 우리는 커서를 기준으로 왼쪽과 오른쪽의 스택을 각각 나눠준다고 생각하자.

l 라는 변수에 문자열을 받고, left와 right를 칭하는 스택을 리스트로 만들어주었다. 그리고 '<', '>', '-'으로 경우를 나누어 해당 기능을 코드로 짜주면 되는데, 나는 에디터 문제처럼 if i == '<' and word_l: 풀었다가 처음에 '틀렸습니다' 가 두 번이나 떴다. 그 이유는, if문이 에디터의 문제와 다르게 if 문을 거치지 못하면 else 문에서 해당 문자가 모두 스택에 append 되기 때문이다. if-else문의 특성상, 가장 마지막에 '<', '>', '-'이 들어갈 수 있기 때문에, 우리는 if 문 내에 if 문 조건을 또 걸어줄 필요가 있다는 것이다.

마지막으로는 join문을 사용해서 왼쪽 스택, 오른쪽 스택을 함께 출력하면 되는데, 스택은 후입선출 방식이기 때문에 reversed상태 로출력해줄 필요가 있다.