[백준] 키로거(5397) - Java

문제

 

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

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

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

---

입력

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

---

출력

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

---

예제 입력 1 

2
<<BP<A>>Cd-
ThIsIsS3Cr3t

예제 출력 1 

BAPC
ThIsIsS3Cr3t

 

 

내가 푼 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.List;
import java.util.ListIterator;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        for (int i=0; i<N; i++) {
            List<Character> list = new LinkedList<>();
            ListIterator listIterator = list.listIterator();
            char[] chars = br.readLine().toCharArray();

            for (int j=0; j<chars.length; j++) {
                switch (chars[j]) {
                    case '>':
                        if (listIterator.hasNext())
                            listIterator.next();
                        break;
                    case '<':
                        if (listIterator.hasPrevious())
                            listIterator.previous();
                        break;
                    case '-':
                        if (listIterator.hasPrevious()) {
                            listIterator.previous();
                            listIterator.remove();
                        }
                        break;
                    default:
                        listIterator.add(chars[j]);
                        break;
                }
            }
            StringBuilder sb = new StringBuilder();
            for(char s : list) {
                sb.append(s);
            }
            System.out.println(sb.toString());
        }
    }
}

 

내가 사용한 방법은 LinkedList 이다.

차례대로 문자를 입력받으면서 LinkedList에 넣어준다.

일반 배열보다 삽입, 삭제 시 더 빠른 자료구조이기 때문에 시과초과되지 않기 위해 사용하였고, 실제 사용할 때 

index를 이용해서 넣으면 LinkedList 를 사용하는 의미가 없어진다.

 

따라서, ListIterator 라는 것을 사용하여 풀었다.

ListIterator 는 아래를 참고하자!

[Java] LinkedList 순회하기 위한 ListIterator 사용하기

 

 

다른 사람이 푼 코드

package org.example.codingtest;

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

public class 키로거 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        for (int i=0; i<N; i++) {
            Stack<Character> left = new Stack<>();
            Stack<Character> right = new Stack<>();

            for (int j=0; j<chars.length; j++) {
                switch (chars[j]) {
                    case '>':
                        if (!right.isEmpty())
                            left.push(right.pop());
                        break;
                    case '<':
                        if (!left.isEmpty())
                            right.push(left.pop());
                        break;
                    case '-':
                        left.pop();
                        break;
                    default:
                        left.add(chars[j]);
                        break;
                }
            }

            if (!right.isEmpty()) {
                left.push(right.pop());
            }

            StringBuilder sb = new StringBuilder();
            for (Character c : left) {
                sb.append(c);
            }
            System.out.println(sb);
        }
    }
}

다른 사람들은 어떤식으로 풀었나 찾아봤더니 Stack을 2개 사용하였다.

조건

1. 커서를 기준으로 왼쪽과 오른쪽을 나누어, 입력/삭제되는 글자는 모두 왼쪽 stack에서 push/pop 한다.

2. 커서가 '<' 이면 왼쪽 stack의 마지막 글자를 pop()해서 오른쪽 stack에 push하고,

3. 커서가 '>' 이면 오른쪽 stack의 마지막 글자를 pop()해서 왼쪽  stack에 push 한다.

 

모두 진행 후 오른쪽 stack에 글자가 있으면 pop해서 왼쪽 stack에 push해주고 출력할때는 왼쪽 stack에서 pop()으로 글자를 하나씩 빼주면 역순이기 때문에 for문으로 순회하여 글자를 StringBuilder에 append() 시킨다.