[ BOJ ] 5397 : 키로거 ( SILVER 2 ) / C, Python

문제

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

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

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

입력

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

출력

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

풀이 과정

스택 두 개를 이용해 유동적으로 문자열 데이터를 처리하는 문제이다.

일반적인 대문자, 소문자, 숫자가 입력되면 스택에 차례대로 입력한다.

만약, 왼쪽 화살표를 두 번 입력했으면 현재 커서는 A 바로 뒤에 있으므로, 왼쪽 화살표를 한 번씩 누르면서 지나간 C 와 B는 차례대로 스택에서 빼서 다른 스택 (화살표 스택) 에 저장해둔다. 커서는 A|BC 와 같아진다.

커서 바로 뒤에 D 와 E 를 입력해 A|BC 에서 ADE|BC 를 만든다.

오른쪽 화살표를 한 번 누르면 왼쪽 화살표로 인해 다른 스택에 저장되어 있던 B가 다시 문자열 스택으로 돌아온다. ADE|BC 에서 ADEB|C 가 된 것이다. 오른쪽 화살표를 끝까지 누른다면 스택이 비게 되어 현재까지 입력한 문자열의 맨 끝으로 가게 된다.

백 스페이스를 두 번 입력하면, 커서가 가리키는 문자가 두 개 사라지고 ADEB|C 에서 AD|C 가 된다.

이렇게 남게 된 문자열은 ADC 가 된다. 

 

요약하자면, 왼쪽 화살표를 입력하면 커서로 넘긴 문자를 다른 스택에 저장하고, 오른쪽 화살표를 입력하면 넘긴 문자를 다시 가져온다. 백 스페이스를 입력하면 현재 커서 앞에 있는 문자를 (문자열 스택 맨 끝에 있는 문자를) 지운다.

C

#include <stdio.h>
#include <string.h>
#define SIZE 1000005

char str[SIZE];
char stack1[SIZE];
int stack1_size = 0;
char stack2[SIZE];
int stack2_size = 0;

int main(void) {
    int t;
    scanf("%d", &t);
    while (t--) {
        scanf("%s", str);
        stack1_size = 0;
        stack2_size = 0;
        for (int i = 0; i < strlen(str); i++) {
            if (str[i] == '<') {
                if (stack1_size != 0) {
                    stack2[stack2_size++] = stack1[stack1_size - 1];
                    stack1_size -= 1;
                }
            }
            else if (str[i] == '>') {
                if (stack2_size != 0) {
                    stack1[stack1_size++] = stack2[stack2_size - 1];
                    stack2_size -= 1;
                }
            }
            else if (str[i] == '-') {
                if (stack1_size != 0) stack1_size -= 1;
            }
            else {
                stack1[stack1_size++] = str[i];
            }
        }

        for (int i = 0; i < stack1_size; i++) printf("%c", stack1[i]);
        for (int i = stack2_size - 1; i >= 0; i--) printf("%c", stack2[i]);
        printf("\n");
    }
    return 0;
}

Python

import sys
input = sys.stdin.readline

test = int(input().rstrip())
for t in range(test):
    string = input().rstrip()
    left_stack = []
    right_stack = []
    for s in string:
        if s == '<':
            if len(left_stack) != 0:
                right_stack.append(left_stack.pop())
        elif s == '>':
            if len(right_stack) != 0:
                left_stack.append(right_stack.pop())
        elif s == '-':
            if len(left_stack) != 0:
                left_stack.pop()
        else:
            left_stack.append(s)
    
    while right_stack:
        left_stack.append(right_stack.pop())
    
    print(''.join(left_stack))