[백준/BOJ] 5397번: 키로거

https://www.acmicpc.net/problem/5397

5397번: 키로거

문제

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

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

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

해설

앞서 풀었었던 1406번과 거의 같은 유형의 문제이다.

https://jjunsu.tistory.com/115?category=979846 

[백준/BOJ] 1406번: 에디터

 

이 문제에서 또한 LinkedList를 이용하면 add, remove의 명령에 O(N)이라는 시간복잡도를 갖는다.

이 때문에 시간 초과가 나기 때문에 ListIterator를 사용하였다.

 

ListIterator에선 add나 remove를 할 때 처음부터 탐색하는 것이 아닌 그 위치에서 명령을 수행한다.

따라서 O(1)의 시간복잡도를 가진다.

 

또한 출력을 할 때

for(i in list.indices) print(list[i])

위와 같이 한다면 시간복잡도 O(N)이 걸린다.

이를 해결하기 위해 toCharArray를 사용하여 반복문을 쓰지 않고 출력을 한다.

소스 코드

import java.util.*

fun main() = with(System.`in`.bufferedReader()) {
    val n = readLine().toInt()
    repeat(n) {
        val sentence = readLine()
        val list = LinkedList<Char>()
        val iterator = list.listIterator()
        for(i in sentence.indices) {
            var c = sentence[i]
            if(c == '<') {
                if(iterator.hasPrevious()) iterator.previous()
            }
            else if(c == '>') {
                if(iterator.hasNext()) iterator.next()
            }
            else if(c == '-') {
                if(iterator.hasPrevious()) {
                    iterator.previous()
                    iterator.remove()
                }
            }
            else iterator.add(c)
        }
        println(list.toCharArray())
    }
}