백준 1406 [kotlin]

문제

한 줄로 된 간단한 에디터를 구현하려고 한다. 이 편집기는 영어 소문자만을 기록할 수 있는 편집기로, 최대 600,000글자까지 입력할 수 있다.

이 편집기에는 '커서'라는 것이 있는데, 커서는 문장의 맨 앞(첫 번째 문자의 왼쪽), 문장의 맨 뒤(마지막 문자의 오른쪽), 또는 문장 중간 임의의 곳(모든 연속된 두 문자 사이)에 위치할 수 있다. 즉 길이가 L인 문자열이 현재 편집기에 입력되어 있으면, 커서가 위치할 수 있는 곳은 L+1가지 경우가 있다.

이 편집기가 지원하는 명령어는 다음과 같다.

L	커서를 왼쪽으로 한 칸 옮김 (커서가 문장의 맨 앞이면 무시됨)
D	커서를 오른쪽으로 한 칸 옮김 (커서가 문장의 맨 뒤이면 무시됨)
B	커서 왼쪽에 있는 문자를 삭제함 (커서가 문장의 맨 앞이면 무시됨) 삭제로 인해 커서는 한 칸 왼쪽으로 이동한 것처럼 나타나지만, 실제로 커서의 오른쪽에 있던 문자는 그대로임
P $	$라는 문자를 커서 왼쪽에 추가함

초기에 편집기에 입력되어 있는 문자열이 주어지고, 그 이후 입력한 명령어가 차례로 주어졌을 때, 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 구하는 프로그램을 작성하시오. 단, 명령어가 수행되기 전에 커서는 문장의 맨 뒤에 위치하고 있다고 한다.

입력

첫째 줄에는 초기에 편집기에 입력되어 있는 문자열이 주어진다. 이 문자열은 길이가 N이고, 영어 소문자로만 이루어져 있으며, 길이는 100,000을 넘지 않는다. 둘째 줄에는 입력할 명령어의 개수를 나타내는 정수 M(1 ≤ M ≤ 500,000)이 주어진다. 셋째 줄부터 M개의 줄에 걸쳐 입력할 명령어가 순서대로 주어진다. 명령어는 위의 네 가지 중 하나의 형태로만 주어진다.

출력

첫째 줄에 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 출력한다.

코드 1 - ITERATOR 사용 

import java.util.LinkedList

private fun main() = with(System.`in`.bufferedReader())
{
    // 데이터 입력 받기
    var inputString = LinkedList(readLine().toCharArray().toList())
    var it = inputString.listIterator()
    while (it.hasNext()) it.next()
    var M = readLine()!!.toInt()
    for (i in 0..M - 1){
        var command = readLine()
        if(command[0] == 'P'){
            it.add(command[2])
        }
        else if(command[0] == 'L'){
            if(it.hasPrevious()) it.previous()
        }
        else if(command[0] == 'D'){
            if(it.hasNext()) it.next()
        }
        else{
            if(it.hasPrevious()) {
                it.previous()
                it.remove()
            }
        }
    }
    println(inputString.joinToString(""))
}

코드 2  [시간 초과] - INDEX값을 이용한 접근 

private fun main()
{
    // 데이터 입력 받기
    var inputString = java.lang.StringBuilder()
    inputString.append(readLine()!!)
    var M = readLine()!!.toInt()
    var cursor:Int = inputString.length
    for (i in 0..M - 1){
        var command = readLine()!!.split(' ')
        if(command[0] == "P"){
            inputString.insert(cursor, command[1])
            cursor++
        }
        else if(command[0] == "L"){
            if(cursor != 0) cursor--
        }
        else if(command[0] == "D"){
            if(cursor != inputString.length) cursor++
        }
        else{
            if(cursor != 0) {
                inputString.deleteCharAt(cursor-1)
                cursor--
            }
        }
    }
    println(inputString)
}

코드 3 - STACK 사용

private fun main() = with(System.`in`.bufferedReader())
{
    var inputString = readLine().toList()
    var lStack = Stack<Char>()
    var rStack = Stack<Char>()
    for (i in inputString)
    {
        lStack.add(i)
    }
    var M = readLine()!!.toInt()
    for (i in 0..M-1)
    {
        var command = readLine()
        if(command[0] == 'P'){
            lStack.add(command[2])
        }
        else if (command[0] == 'L'){
            if(lStack.size != 0) rStack.add(lStack.pop())
        }
        else if (command[0] == 'D'){
            if(rStack.size != 0) lStack.add(rStack.pop())
        }
        else{
            if (lStack.size != 0) lStack.pop()
        }
    }
    lStack.addAll(rStack.reversed())
    println(lStack.joinToString(""))
}

후기

string을 substring을 사용해서 인덱싱을 하는 것처럼 index를 활용하여 배열에 접근하는 방식은 시간 복잡도가 o(n)이기에 시간이 너무 오래 걸려 시간초과가 발생했다. 그래서 iterator를 사용하여 해결하였다. iterator를 사용하게 되면 추가 삭제가 시간 복잡도가 o(1)이 되므로 더 빠르다. stack을 사용하여 푸는 방법도 있다고 해서 그 방법으로도 풀어보았다.. stack을 사용해서 stack + 커서 + stack 이런 방식으로 풀어나가는 방식이었다. 커서가 L이나 D에 의해 옮겨지면 stack의 pop과 push를 이용하여 값을 빼고 추가해서 커서를 이동시키는 방식이라고 한다.