hello-algo/codes/kotlin/chapter_computational_complexity/complexity_exercises.kt

/**
 * File: complexity_exercises.kt
 * Created Time: 2026-08-18
 * Author: Hello Algo Team
 */

package chapter_computational_complexity.complexity_exercises

/* 迭代求和 */
fun sumIter(n: Int): Int {
    var res = 0
    for (i in 1..n) {
        res += i
    }
    return res
}

/* 递归求和 */
fun sumRecur(n: Int): Int {
    if (n == 1) {
        return 1
    }
    return n + sumRecur(n - 1)
}

/* 线性阶循环 */
fun linearLoop(n: Int): Int {
    var res = 0
    for (i in 0 until n) {
        res += i
    }
    return res
}

/* 平方阶循环 */
fun quadraticLoop(n: Int): Int {
    var res = 0
    for (i in 0 until n) {
        for (j in i until n) {
            res += j
        }
    }
    return res
}

/* 对数阶循环 */
fun logarithmicLoop(n: Int): Int {
    var value = n
    while (value > 1) {
        value /= 2
    }
    return value
}

fun main() {
    check(sumIter(1) == 1 && sumRecur(1) == 1)
    check(sumIter(4) == 10 && sumRecur(4) == 10)
    check(linearLoop(4) == 6)
    check(quadraticLoop(4) == 20)
    check(logarithmicLoop(4) == 1 && logarithmicLoop(5) == 1)
}