[BOJ] 5397 - 키로거 JAVA

www.acmicpc.net/problem/5397

5397번: 키로거

문제

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

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

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

입력

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

출력

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

예제 입력 1

2

<<BP<A>>Cd-

ThIsIsS3Cr3t

예제 출력 1

BAPC

ThIsIsS3Cr3t

 

 

 

 

 

 

 

풀이 1. (틀린 코드 - 시간 초과)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder ans = new StringBuilder();

        int T = Integer.parseInt(br.readLine());
        while(T-- > 0) {
            StringBuilder sb = new StringBuilder();
            String str = br.readLine();
            int len = str.length();

            int cursor = 0;
            for(int i = 0; i < len; i++) {
                char ch = str.charAt(i);
                if(ch == '<') {
                    if(cursor != 0)
                        cursor -= 1;
                }else if(ch == '>') {
                    if(cursor != sb.length())
                        cursor += 1;
                }else if(ch == '-') {
                    if(cursor != 0) {
                        sb.deleteCharAt(cursor-1);
                        cursor -= 1;
                    }
                }else {  // 그냥 문자
                    sb.insert(cursor, ch);
                    cursor += 1;
                }
            }
            ans.append(sb.toString()).append("\n");
        }
        System.out.println(ans.toString());
    }
}

 

StringBuilder를 사용한 시간 초과 풀이. 50%에서 시간 초과가 발생한다.

입력의 길이가 최대 1,000,000이라는 점이 좀 불안하긴 했지만 괜찮을 거라고 생각했는데 아니었다.

 

sb.insert()는 System.arraycopy()를 매번 호출하는데 이것 때문에 O(n)의 시간이 소요된다.

sb.append()는 System.arraycopy()를 호출하긴 하지만 용량을 다 사용하면 한 번 호출하여 두배를 늘리는 식으로, 매번 호출되는 것이 아니기 때문에 O(1)로 해결이 가능하다.

 

위 코드에서는 sb.insert()를 너무 많이 사용해서 시간 초과가 난 듯 하다.

 

 

 

 

 

 

풀이 2. (정답 코드) 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        int T = Integer.parseInt(br.readLine());
        while(T-- > 0) {
            Stack<Character> left = new Stack<>();
            Stack<Character> right = new Stack<>();
            String str = br.readLine();

            int len = str.length();
            for(int i = 0; i < len; i++) {
                char ch = str.charAt(i);

                if(ch == '<') {
                    if(!left.isEmpty()) {
                        right.push(left.pop());
                    }
                }else if(ch == '>') {
                    if(!right.isEmpty()) {
                        left.push(right.pop());
                    }
                }else if(ch == '-') {
                    if(!left.isEmpty()) {
                        left.pop();
                    }
                }else {  // 그냥 문자
                    left.push(ch);
                }
            }

            while(!left.isEmpty()) right.push(left.pop());
            while(!right.isEmpty()) sb.append(right.pop());
            sb.append("\n");
        }
        System.out.println(sb.toString());
    }
}

 

스택을 이용한 풀이.

 

어느 문제였는지 기억은 안 나지만 똑같은 방식을 사용한 문제를 푼 적이 있었다.