[백준] 5397번 키로거 (Python)

5397번: 키로거

문제

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

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

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

입력

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

출력

 

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

과정

• 커서를 기준으로 왼쪽은 left라는 배열에 , 오른쪽은 right라는 배열에 담아 배열2개를 활용하여 풀어준다.

1. 커서를 왼쪽으로 움직이는경우 : 왼쪽의 마지막 원소를 오른쪽리스트에 추가

2. 커서를 오른쪽으로 움직이는 경우 : 오른쪽 마지막 원소를 왼쪽리스트에 추가

3. 문자가 추가되는경우 : 왼쪽리스트에 추가 (스택처럼 쌓임)

4. 문자가 삭제되는경우 : 왼쪽리스트의 마지막원소를 제거

• 커서를 왼쪽으로 움직여서 오른쪽리스트가 추가될때 스택형식으로 쌓이므로 오른쪽 리스트에는 실제 문자열의 정렬과 반대로 정렬되어 쌓인다. 
• 그러므로 리스트를 이어줄 때 뒤집어서 이어준다.

 

예제 입력

2
<<BP<A>>Cd-
ThIsIsS3Cr3t

예제 출력

BAPC
ThIsIsS3Cr3t

 

풀이

import sys
input = sys.stdin.readline

n = int(input())

for _ in range(n):
    pwd = list(input().rstrip())
    left, right = [], []

    for i in pwd:
        if i == "<":
            if left:
                right.append(left.pop())
        elif i == ">":
            if right:
                left.append(right.pop())
        elif i == "-":
            if left:
                left.pop()
        else:
            left.append(i)

    left.extend(reversed(right))
    print("".join(left))

 

한 줄씩 해석해보기

import sys
input = sys.stdin.readline

n = int(input())

• 시간 지연 방지를 위한 sys.stdin.readline을 사용한다.
• 테스트 케이스 개수를 입력받는다.

for _ in range(n):
    pwd = list(input().rstrip())
    left, right = [], []

• 테스트 케이스 개수만큼 반복문을 돌린다.
• 패스워드를 입력받는다.
• rstrip() : 이어진 글자를 하나씩 잘라서 받음
• 문자를 입력 받을 리스트를 만들어준다.

for i in pwd:
    if i == "<":
        if left:
            right.append(left.pop())

• 만약 문자가 "<" 라면 즉, 커서를 왼쪽으로 움직이는 경우라면
• left 리스트에 값이 있으면 왼쪽의 마지막 원소를 오른쪽 리스트에 추가한다.

elif i == ">":
    if right:
        left.append(right.pop())

• 만약 문자가 ">"라면 즉, 커서를 오른쪽으로 움직이는 경우라면
• right 리스트에 값이 있으면 오른쪽 마지막 원소를 왼쪽 리스트에 추가한다.

elif i == "-":
    if left:
        left.pop()

• 만약 문자가 "-"라면 
• left 리스트의 마지막 원소를 제거해준다.

else:
    left.append(i)

• 그 외 문자가 입력되면 left 리스트에 추가한다.

left.extend(reversed(right))

• while문이 끝나면 left 리스트에 right리스트를 reverse 해준 후 이어준다.
• 커서를 왼쪽으로 움직여서 오른쪽리스트가 추가될때 스택형식으로 쌓이므로 오른쪽 리스트에는 실제 문자열의 정렬과 반대로 정렬되어 쌓이기 때문이다.

print("".join(left))

• 문자 형식으로 출력해준다.