Kotlinのfold・reduceを使った累積計算テクニック
生徒
「Kotlinで合計や文字の連結みたいな累積計算をする方法ってありますか?」
先生
「はい、Kotlinではfoldとreduceという関数を使えば、コレクションの要素を順番に処理して累積的な結果を作ることができますよ。」
生徒
「どっちも似てそうですが、違いはあるんですか?」
先生
「あります。foldは初期値を自分で決められる、reduceは最初の要素を初期値として使うという違いです。では詳しく見ていきましょう。」
1. foldの基本:初期値を決めて計算をスタートする
fold(フォールド)は、日本語で「折りたたむ」や「たたみ込み」という意味です。バラバラにあるリストの要素を、一つの結果にギュッとまとめ上げるイメージですね。
最大の特徴は、計算を始めるための「初期値」を自由に設定できることです。たとえば、0から計算を始めるのか、100から始めるのかを自分で選べます。
まずは、プログラミング未経験の方でもイメージしやすいように、4つの数字が入った箱(リスト)を全部足す例を見てみましょう。
fun main() {
// 1. 計算したい数字のリストを用意
val numbers = listOf(1, 2, 3, 4)
// 2. foldを使って合計を出す(初期値は0)
val sum = numbers.fold(0) { acc, num ->
println("現在の合計(acc): $acc, 次に足す数字(num): $num")
acc + num
}
println("最終的な答え: $sum")
}
現在の合計(acc): 0, 次に足す数字(num): 1
現在の合計(acc): 1, 次に足す数字(num): 2
現在の合計(acc): 3, 次に足す数字(num): 3
現在の合計(acc): 6, 次に足す数字(num): 4
最終的な答え: 10
このプログラムで登場する2つのキーワードを覚えておくと、一気に理解が深まります。
- acc(アキュムレータ):これまでの計算結果を貯めておく「合計ボックス」のようなものです。最初は指定した初期値(今回は0)から始まります。
- num(要素):リストから順番に取り出される「今の数字」です。
foldの便利なポイントは、もしリストが空(中身が何もない状態)であっても、エラーにならずに最初に決めた初期値をそのまま返してくれることです。そのため、システム開発の現場では、予期せぬエラーを防げる「安全な計算方法」として非常によく使われています。
2. reduceの基本
reduceはfoldに似ていますが、初期値を与えずに最初の要素を初期値として使います。
fun main() {
val numbers = listOf(1, 2, 3, 4)
val sum = numbers.reduce { acc, num -> acc + num }
println(sum)
}
10
ただし、空のリストで実行するとエラーになるため、必ず要素が1つ以上あることを確認してから使う必要があります。
3. foldとreduceの違い
- fold → 初期値を自由に設定でき、空リストでも安全
- reduce → 初期値なしで最初の要素を利用、空リストはエラー
安全性を重視するならfold、コードを短く書きたい場合はreduceを選びましょう。
Kotlinを基礎からしっかり学びたい人や、 Java経験を活かしてモダンな言語にステップアップしたい人には、 定番の入門書がこちらです。
基礎からわかるKotlinをAmazonで見る※ Amazon広告リンク
4. 文字列の連結にも使える
数値だけでなく文字列を累積して結合することもできます。
fun main() {
val words = listOf("Kotlin", "is", "fun")
val sentence = words.fold("") { acc, word -> "$acc $word" }.trim()
println(sentence)
}
Kotlin is fun
このように、累積計算は数値計算だけでなく文字列操作にも応用できます。
5. 条件付き計算の例
例えば、偶数だけを足し合わせたい場合はfilterと組み合わせます。
fun main() {
val numbers = listOf(1, 2, 3, 4, 5, 6)
val evenSum = numbers.filter { it % 2 == 0 }.reduce { acc, num -> acc + num }
println(evenSum)
}
12
このように、事前に条件で絞り込みをしてから累積処理を行うことで、柔軟な集計が可能です。
6. 実用例:買い物かごの合計金額
買い物かごの中の商品合計金額を計算する例です。
data class Item(val name: String, val price: Int)
fun main() {
val cart = listOf(
Item("りんご", 120),
Item("バナナ", 80),
Item("牛乳", 150)
)
val total = cart.fold(0) { acc, item -> acc + item.price }
println(total)
}
350
このように、foldやreduceを使うと複雑な計算処理を簡潔に書くことができます。