Curso Kotlin Puzzle 02

curso-kotlin-puzzle02.kt

import kotlin.collections.List

fun List<Int>.produto(): Int {
    var product = 1
    for (element in this) {
        product *= element
    }
    return product
}

fun processarDespesas(despesas: List<Int>, numeroEntradas: Int): Int? {
    val combinations = despesas.combinacao(numeroEntradas)
    for (combination in combinations) {
        if (combination.sum() == 2020) {
            return combination.produto()
        }
    }
    return null
}

fun <T> List<T>.combinacao(numeroEntradas: Int): Sequence<List<T>> {
    if (numeroEntradas > size) return emptySequence()
    if (numeroEntradas == 0) return sequenceOf(emptyList())
    return drop(1).combinacao(numeroEntradas - 1).map { listOf(first()) + it } + drop(1).combinacao(numeroEntradas)
}

fun main() {
    val despesas = listOf(1721, 979, 366, 299, 675, 1456)
    var numeroEntradas = 2
    var resultado = processarDespesas(despesas, numeroEntradas)
    println("Produto de $numeroEntradas entradas = $resultado")

    numeroEntradas = 3
    resultado = processarDespesas(despesas, numeroEntradas)
    println("Produto de $numeroEntradas entradas = $resultado")
}