BOJ 5397

2020-05-15

위 문제는 백준 사이트의 알고리즘 5397 문제에 관한 설명입니다.

문제

창영이는 강산이의 비밀번호를 훔치기 위해서 강산이가 사용하는 컴퓨터에 키로거를 설치했다.

며칠을 기다린 끝에 창영이는 강산이가 비밀번호 창에 입력하는 글자를 얻어냈다.

키로거는 사용자가 키보드를 누른 명령을 모두 기록한다.

따라서, 강산이가 비밀번호를 입력할 때, 화살표나 백스페이스를 입력해도 정확한 비밀번호를 알아낼 수 있다.

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

입력

첫째 줄에 테스트 케이스의 개수가 주어진다.

각 테스트 케이스는 한줄로 이루어져 있고, 강산이가 입력한 순서대로 길이가 L인 문자열이 주어진다.

(1 ≤ L의 길이 ≤ 1,000,000) 강산이가 백스페이스를 입력했다면, ‘-‘가 주어진다.

이때 커서의 바로 앞에 글자가 존재한다면, 그 글자를 지운다. 화살표의 입력은 ‘<’와 ‘>’로 주어진다.

이때는 커서의 위치를 움직일 수 있다면, 왼쪽 또는 오른쪽으로 1만큼 움직인다. 나머지 문자는 비밀번호의 일부이다.

물론, 나중에 백스페이스를 통해서 지울 수는 있다.

만약 커서의 위치가 줄의 마지막이 아니라면, 그 문자를 입력하고, 커서는 오른쪽으로 한 칸 이동한다.

출력

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

import java.util.Scanner;
import java.util.Stack;

class Key {
    public static void main(String[] args) {
        Scanner scanner= new Scanner(System.in);
        int testCase = scanner.nextInt();//testcase받기
        for(int tc = 0; tc < testCase; tc++)
        {
            String str = scanner.next();//문자열 입력
            Stack<Character> stack1 = new Stack<>();//문자 하나하나씩 stack에 넣을 예정
            Stack<Character> stack2 = new Stack<>();//"<"를 통해 빼낸 문자열 저장할 stack 
            for(int i = 0; i < str.length();i++)
            {
                if(str.charAt(i)=='<')
                {
                    if(!stack1.isEmpty()){
                        stack2.push(stack1.pop());//stack1에 값이 들어있으므로 빼내어서 stack2에 임시 저장
                    }
                }
                else if(str.charAt(i)=='>')
                {
                    if(!stack2.isEmpty()){
                        stack1.push(stack2.pop());//stack2에 값이 들어있으므로 빼내어서 stack1에 임시 저장
                    }
                }
                else if(str.charAt(i)=='-')
                {
                    if(!stack1.isEmpty()){
                        stack1.pop();//-는 삭제이므로 stack1에 저장된 값을 하나 빼줌
                    }
                }
                else{
                    stack1.push(str.charAt(i));//나머지는 그대로 입력시키면 됨
                }
            }
            while(!stack2.isEmpty())//stack2에 값이있다면?
            {
                stack1.push(stack2.pop());//남은 값들을 넣어줍니다.
            }
            StringBuilder sb = new StringBuilder();//써주는 이유는 일반 println으로 하나씩 하면 시간초과
            while(!stack1.isEmpty())
                sb.append(stack1.pop()); // remove를 써서 앞에서부터 지워봤지만 시간초과가 남 : 이유로는 remove를 하면
                                         // 삭제 후 앞으로 땡겨오면서 다시 정렬시키게 됨으로 시간이 더 올라간다.
            System.out.println(sb.reverse()); // 그래서 역순으로 뺀 후 reverse로 한번에 정렬하는게 더 빠름
        }
    }
}

설명은 주석 참고!

키로거