[백준] 수열의 변화 (1551)(kotlin)

문제 설명

백준 1551번 문제 링크

입력 및 출력

» 입력

첫째 줄에 수열의 크기 N과 K가 주어진다. N은 20보다 작거나 같은 자연수이고, K는 0보다 크거나 같고, N-1보다 작거나 같은 자연수이다. 둘째 줄에는 수열이 ‘,’로 구분되어 주어진다.

» 출력

첫째 줄에 K번 변형한 수열을 ‘,’로 구분하여 출력한다.

예제 입출력(테스트케이스)

입력 출력
5 1
5,6,3,9,-1
1,-3,6,-10

문제 풀이1

import java.io.BufferedReader
import java.io.InputStreamReader

fun main(args: Array<String>) = with(BufferedReader(InputStreamReader(System.`in`))) {
    val (N, K) = readLine().split(" ").map { it.toInt() }
    var count = 0
    var seq = readLine().split(",").map { it.toInt() }

    while (count != K) {
        count++
        seq = List(seq.size - 1) { seq[it + 1] - seq[it] }
    }

    println(seq.joinToString(","))
}