hello-algo/ko/codes/kotlin/chapter_computational_complexity/iteration.kt

/**
 * File: iteration.kt
 * Created Time: 2024-01-25
 * Author: curtishd ([email protected])
 */

package chapter_computational_complexity.iteration

/* for 반복문 */
fun forLoop(n: Int): Int {
    var res = 0
    // 반복문으로 1, 2, ..., n-1, n의 합을 구한다
    for (i in 1..n) {
        res += i
    }
    return res
}

/* while 반복문 */
fun whileLoop(n: Int): Int {
    var res = 0
    var i = 1 // 조건 변수를 초기화한다
    // 반복문으로 1, 2, ..., n-1, n의 합을 구한다
    while (i <= n) {
        res += i
        i++ // 조건 변수를 갱신한다
    }
    return res
}

/* while 반복문(두 번 갱신) */
fun whileLoopII(n: Int): Int {
    var res = 0
    var i = 1 // 조건 변수를 초기화한다
    // 반복문으로 1, 4, 10, ...의 합을 구한다
    while (i <= n) {
        res += i
        // 조건 변수를 갱신한다
        i++
        i *= 2
    }
    return res
}

/* 이중 for 반복문 */
fun nestedForLoop(n: Int): String {
    val res = StringBuilder()
    // i = 1, 2, ..., n-1, n을 반복한다
    for (i in 1..n) {
        // j = 1, 2, ..., n-1, n을 반복한다
        for (j in 1..n) {
            res.append(" ($i, $j), ")
        }
    }
    return res.toString()
}

/* 실행 코드 */
fun main() {
    val n = 5
    var res: Int

    res = forLoop(n)
    println("\nfor 반복문으로 구한 합 res = $res")

    res = whileLoop(n)
    println("\nwhile 반복문으로 구한 합 res = $res")

    res = whileLoopII(n)
    println("\nwhile 반복문(두 번 갱신)으로 구한 합 res = $res")

    val resStr = nestedForLoop(n)
    println("\n이중 for 반복문의 순회 결과 $resStr")
}