[백준][17413] 단어 뒤집기2

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

17413번: 단어 뒤집기 2

문제 풀이

1. 문제 조건을 문자열의 char 를 하나씩 탐색하여 구현할 경우 시간 초과가 발생한다.

2. 문자열을 검색할 때 첫 문자가 "<" 로 시작되는 지 파악 한다.

    2-1. "<" 로 시작될 경우 indexOfFirst 를 이용하여 ">" index 를 구한다.

    2-2. "<" 와 ">" 로 되어 있는 문자열은 그대로 result에 저장한다.

    2-3. result 에 저장한 문자열 다음 문자 부터 다시 탐색한다.

3. 문자열을 검색할 때 "<" 로 시작되지 않는 지 파악 한다.

    3-1. 뒤에 "<" 문자가 나오는 지 indexOfFirst 를 이용하여 확인한다.

    3-2. 3-1 의 결과가 -1일 경우 "<" 문자가 더 이상 없으므로 현재 남은 문자열을 공백을 기준으로 짤라서 reverse 처리하여 result 에 저장한다.

     3-3. 3-1 의 결과가 -1이 아닐 경우 "<" 문자가 존재하기 때문에 3-1의 index 전까지 문자열을 잘라서 공백을 기준으로 revese 처리하여 result 에 저장한다.

     3-4. 남은 문자열은 2번 과정을 다시 수행한다.

4. len 값이 0이 될 때까지 반복한다.

코드

import java.io.*
import java.util.*

fun main(args: Array<String>) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))
    var str = br.readLine()
    val sb = StringBuilder()
    var result = ""
    var len = str.length

    while (len != 0) {
        //println("str = $str")
        if (str[0] == '<') {
            val idx = str.indexOfFirst {
                it == '>'
            }

            result += str.substring(0, idx + 1)
            str = str.substring(idx + 1, len)
            len = str.length
            //println("<> str idx = $idx, len = $len str = $str, result = $result")
        } else {
            val idx = str.indexOfFirst {
                it == '<'
            }

            // 뒤에 더 이상 < 문자가 없는 문자 열이다
            // 공백을 기준으로 짤라서 짤려진 문자를 reverse 하여 result 에 저장한다.
            if (idx == -1) {
                val list = str.split(" ").map { it.reversed() }

                list.forEachIndexed { index, s ->
                    result += if (index == 0) {
                        s
                    } else {
                        " $s"
                    }
                }

                break
            } else {
                val list = str.substring(0, idx).split(" ").map { it.reversed() }

                list.forEachIndexed { index, s ->
                    result += if (index == 0) {
                        s
                    } else {
                        " $s"
                    }
                }

                str = str.substring(idx, len)
                len = str.length

                //println("222 <> str idx = $idx, len = $len str = $str, result = $result")
            }
        }
    }


    //println(result)
    bw.write(result)
    bw.flush()
    bw.close()
    br.close()
}