Kotlin KoansでKotlin入門 第２８回：Sum

はじめに

公式の問題集「Kotlin Koans」を解きながらKotlinを学習します。

過去記事はこちら

• Introduction
  • 第０回：Kotlin Koansについて
  • 第１回：Simple Functions
  • 第２回：Named arguments
  • 第３回：Default arguments
  • 第４回：Triple-quoted strings
  • 第５回：String templates
  • 第６回：Nullable types
  • 第７回：Nothing type
  • 第８回：Lambdas
• Classes
  • 第９回：Data classes
  • 第１０回：Smart casts
  • 第１１回：Sealed classes
  • 第１２回：Rename on import
  • 第１３回：Extension functions
• Conventions
  • 第１４回：Comparison
  • 第１５回：Ranges
  • 第１６回：For loop
  • 第１７回：Operators overloading
  • 第１８回：Invoke
• Collections
  • 第１９回：Collections Introduction
  • 第２０回：Sort
  • 第２１回：Filter map
  • 第２２回：All, Any, and other predicates
  • 第２３回：Associate
  • 第２４回：Group By
  • 第２５回：Partition
  • 第２６回：FlatMap
  • 第２７回：Max min

問題

Sum

ある顧客が注文したすべての商品の価格の合計である、顧客が費やした金額の合計を計算する関数を実装しなさい。
各製品は注文された回数だけカウントされることに注意せよ。

修正前コード.kt

// Return the sum of prices for all the products ordered by a given customer
fun moneySpentBy(customer: Customer): Double =
        TODO()

Shop.kt

data class Shop(val name: String, val customers: List<Customer>)

data class Customer(val name: String, val city: City, val orders: List<Order>) {
    override fun toString() = "$name from ${city.name}"
}

data class Order(val products: List<Product>, val isDelivered: Boolean)

data class Product(val name: String, val price: Double) {
    override fun toString() = "'$name' for $price"
}

data class City(val name: String) {
    override fun toString() = name
}

問題のポイント

数値の集まりにsumを使うか、sumOfを使って、まず要素を数値に変換してから合計します。

listOf(1, 5, 3).sum() == 9
listOf("a", "b", "cc").sumOf { it.length } == 4

ProductのpriceをsumOfを使って合計するために、ordersでそれぞれ保持しているproductsをflatMapを使って1つのリストにまとめます。

flatmap.kt

val list = listOf("123", "45")
println(list.flatMap { it.toList() }) // [1, 2, 3, 4, 5]

解答例

fun moneySpentBy(customer: Customer): Double =
        customer.orders.flatMap { it.products }.{ it.price }