[Python] S2_5397_키로거⌨

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

문제

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

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

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

입력

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

출력

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

조건

시간 제한: 1초
메모리 제한: 256MB

접근

1. for문 돌려서 하나씩 받으면서 커서를 이동할지, 제거할지, 추가할지 체크하자
2. 현재 인덱스를 계속 움직이면서, 즉 커서를 직접 움직이면서 추가 혹은 제거하면 되는 문제 아닌가?

라는 생각으로 짠 코드가 아래의 코드이다.

초기 코드

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    W = input().rstrip()
    lst = []
    current_i = 0
    for i in W:
        if i == '<' and lst and current_i != 0:      # 좌측 커서 이동
            current_i -= 1
        
        elif i == '>' and lst and current_i < len(lst):    # 우측 커서 이동
            current_i += 1
            
        elif i == '-' and lst:    # 직전 제거
            lst.pop(current_i-1)
            current_i -= 1
            
        elif i.isalnum():                   # 문자열
            lst.insert(current_i, i)
            current_i += 1
            
    print(''.join(lst))

이렇게 짜니 대부분의 예제들은 잘 나왔는데, 제출하니 index error가 나왔다. 그리고 이 문제는 입력값이 1,000,000까지라서, 아마 시간초과도 나왔을 듯하다.

찾아 보니 연결 리스트를 사용한다는 데 이 부분은 아직 잘 모르고.. 내 눈높이에 조금 더 맞는 코드를 찾았다.

🔔 바로 커서는 그대로 두고, 리스트를 left, right 두 개로 나누어 보는 것이었다.

이렇게 보면 이해가 잘 안되니 코드와 함께 설명하겠다.

수정된 코드

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    W = input().rstrip()
    left, right = [], []
    
    for s in W:
        # <1. 커서를 왼쪽으로 옮기는 경우>
        if s == '<' and left:
            right.append(left.pop())
        # <2. 커서를 오른쪽으로 옮기는 경우>    
        elif s == '>' and right:
            left.append(right.pop())
        # <3. 제거하는 경우>    
        elif s == '-' and left:
            left.pop()
        # <4. 추가하는 경우>    
        elif s.isalnum():
            left.append(s)
        
    left.extend(reversed(right))         
    print(''.join(left))

말 그대로 커서는 두 리스트 사이에 가만히 있는다고 가정하고, 리스트 안에 있는 원소를 가지고 왔다갔다 하는 것이다.

1. 왼쪽으로 커서를 옮기게 되면, left 가장 끝에 있는 원소를 right로 옮긴다.
2. 오른쪽으로 커서를 옮기게 되면, right 가장 끝에 있는 원소를 left로 옮긴다.
3. 제거하는 경우에는 left 가장 끝 원소를 하나 제거한다.
   (* 커서를 옮기거나 제거하는 경우에서는 당연히 리스트 안에 원소가 있음도 함께 체크해준다.)
4. 문자나 숫자가 나오면, left에 추가해준다.

그리고 마지막으로 중요한 점이, left에서 pop하여 right에 넣어준 것이기 때문에, right의 배열은 기존 문자열과 반대가 된다.

예를 들어 left가 [a,b,c,d]인 상태에서 <가 2번 연속으로 나온다면, right는 [d,c]가 될 것이다.

그렇기에 두 리스트를 합치기 전에, right를 뒤집은 후 합쳐 출력해준다.

느낀 점 & 배운 점

1. 역시 스택을 잘 쓰면 문제를 창의적으로 풀 수 있는 것 같다.. 문제가 해결이 안될 때는 아예 다른 방식으로도 생각을 해 보아야할 것 같다.
2. 아직 실버2도 쉽게 못 푸는 걸 보면 역시 난 멀었다. 더 열심히 하자!