[Java][백준][5397번] 키로거

문제 설명

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

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

강산이가 비밀번호 창에서 입력한 키가 주어졌을 때, 강산이의 비밀번호를 알아내는 프로그램을 작성하시오.

입력

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

출력

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

 

생각

처음엔 일반적으로 switch문을 사용하려 했으나 바로 시간초과가 나버렸다. 게다가 String에 계속적으로 추가하는 방식을 사용하니 역시 시간초과가 나버렸다. 방법을 검색하던 중, 글자의 이동이나 삭제 삽입의 경우에는 스택을 사용하는게 좋다는 말을 들었다. 또한 StringBuilder를 사용하면 시간을 단축할 수 있다는 말을 들었다.

 

스택을 2개 사용하여 <가 나오면 다른 스택에 넣어주고 >가 나올 때 다시 꺼내서 원래 스택에 넣어준다. 이렇게 하면 커서의 위치를 적절하게 조절할 수 있을 것이다. for문을 다 돌고 나면 나머지 스택에 문자가 남아있을 수도 있고 아닐 수도 있으므로 (커서의 위치가 문자열의 끝이 아니었다면 다른 스택에 문자가 남아있을 것이다.) 다른 스택이 isEmpty가 될때 까지 정답 스택에 넣어주면 된다.

 

그리고 StringBuilder을 넣어주면 되는데, 현재 스택에는 순서대로 들어있으므로 pop해서 빼내게 되면 거꾸로 나오게 된다. 따라서 컬렉션 스택의 함수 중 하나인 ElementAt 메서드를 사용하여 아래 있는 요소부터 받아와서 넣어주면 된다.

 

코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;
 
public class Main {
 
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        int t = Integer.parseInt(br.readLine());
        
        for(int j=0; j<t; j++) {
            char[] str = br.readLine().toCharArray();
            
            Stack<Character> a1 = new Stack<Character>();
            Stack<Character> a2 = new Stack<Character>();
            
            for(int i=0; i<str.length; i++) {
                switch(str[i]) {
                case '<':
                    if(!a2.isEmpty()) {
                        a1.push(a2.pop());
                    }
                    break;
                case '>':
                    if(!a1.isEmpty()) {
                        a2.push(a1.pop());
                    }
                    break;
                case '-':
                    if(!a2.isEmpty()) {
                        a2.pop();
                    }
                    break;
                default:
                    if(str[i] != '<' && str[i] != '>' && str[i] != '-') {
                        a2.push(str[i]);
                    }
                    break;
                }
            }
            
            while(!a1.isEmpty()) {
                a2.push(a1.pop());
            }
            
            StringBuilder sb = new StringBuilder();
            
            for(int i=0; i<a2.size(); i++) {
                sb.append(a2.elementAt(i));
            }
            
            System.out.println(sb);
            
        }
    }
}