11656번: 접미사 배열

문제 출처 : https://www.acmicpc.net/problem/11656

11656번: 접미사 배열

언어 : Kotlin

 

문제 설명 :

접미사 배열은 문자열 S의 모든 접미사를 사전순으로 정렬해 놓은 배열이다.

baekjoon의 접미사는 baekjoon, aekjoon, ekjoon, kjoon, joon, oon, on, n 으로 총 8가지가 있고, 이를 사전순으로 정렬하면, aekjoon, baekjoon, ekjoon, joon, kjoon, n, on, oon이 된다.

문자열 S가 주어졌을 때, 모든 접미사를 사전순으로 정렬한 다음 출력하는 프로그램을 작성하시오.

• 첫째 줄에 문자열 S가 주어진다. S는 알파벳 소문자로만 이루어져 있고, 길이는 1,000보다 작거나 같다.
• 첫째 줄부터 S의 접미사를 사전순으로 한 줄에 하나씩 출력한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 256MB

 

입출력 예 :

입력	출력
baekjoon	aekjoon baekjoon ekjoon joon kjoon n on oon

풀이 : 

import java.io.BufferedWriter
import java.io.OutputStreamWriter

fun main() = with(System.`in`.bufferedReader()) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val s = readLine()
    val sb = StringBuilder()
    val arr = mutableListOf<String>()
    repeat(s.length) {
        for (i in it until s.length) { sb.append(s[i]) }
        arr += sb.toString()
        sb.clear()
    }

    arr.sorted().forEach { bw.appendLine(it) }
    bw.flush()
    bw.close()
}