[백준] 점수계산 (2506)(kotlin)

문제 설명

백준 2506번 문제 링크

입력 및 출력

» 입력

첫째 줄에 문제의 개수 N (1 ≤ N ≤ 100)이 주어진다. 둘째 줄에는 N개 문제의 채점 결과를 나타내는 0 혹은 1이 빈 칸을 사이에 두고 주어진다. 0은 문제의 답이 틀린 경우이고, 1은 문제의 답이 맞는 경우이다.

» 출력

첫째 줄에 입력에서 주어진 채점 결과에 대하여 가산점을 고려한 총 점수를 출력한다.

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

입력	출력
10 1 0 1 1 1 0 0 1 1 0	10

문제 풀이1

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

fun main(args: Array<String>) = with(BufferedReader(InputStreamReader(System.`in`))) {
    readLine()

    val result = readLine()
        .split(" ")
        .map { it.toInt() }

    var serises = 0
    var answer = 0

    result.forEach {
        if (it == 1) {
            answer += (1 + serises)
            serises += 1
        }
        else serises = 0
    }

    println(answer)
}