Kotlin KoansでKotlin入門 第３２回：Getting used to the new style

はじめに

公式の問題集「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
  • 第２９回：Fold
  • 第３０回：Compound tasks
  • 第３１回：Sequences

問題

Getting used to the new style

ラムダとコレクションに対する操作を使って、以下のコードを書き換えて簡略化することができます。doSomethingWithCollectionOldStyle関数の簡略版であるdoSomethingWithCollectionの動作が変わらないように変更します。

fun doSomethingWithCollectionOldStyle(
    collection: Collection<String>
): Collection<String>? {
    val groupsByLength = mutableMapOf<Int, MutableList<String>>()
    for (s in collection) {
        var strings: MutableList<String>? = groupsByLength[s.length]
        if (strings == null) {
            strings = mutableListOf()
            groupsByLength[s.length] = strings
        }
        strings.add(s)
    }

    var maximumSizeOfGroup = 0
    for (group in groupsByLength.values) {
        if (group.size > maximumSizeOfGroup) {
            maximumSizeOfGroup = group.size
        }
    }

    for (group in groupsByLength.values) {
        if (group.size == maximumSizeOfGroup) {
            return group
        }
    }
    return null
}

修正前コード.kt

fun doSomethingWithCollection(collection: Collection<String>): Collection<String>? {

    val groupsByLength = collection.groupBy { s -> TODO() }

    val maximumSizeOfGroup = groupsByLength.values.map { group -> TODO() }.maxOrNull()

    return groupsByLength.values.firstOrNull { group -> TODO() }
}

問題のポイント

これまでの復習の問題になります。
doSomethingWithCollectionOldStyle関数で行っていることを確認します。

1. groupsByLength の取得
   • 引数の文字列リストを文字列長でグループ化してgroupsByLengthに設定します。
   • groupsByLengthは、キー：文字列長、値：文字列リストになります。
2. maximumSizeOfGroup の取得
   • groupsByLengthの値で要素数が多い文字列リストをmaximumSizeOfGroupとして取得します。
3. group の返却
   • 要素数が多い文字列リストを返却します

上記を行うために必要になる知識を復習します。

groupBy

第２４回：Group By

val words = listOf("a", "abc", "ab", "def", "abcd")
val byLength: Map<Int, List<String>> = words.groupBy { it.length }
println(byLength.toString()) // {1=[a], 3=[abc, def], 2=[ab], 4=[abcd]}

map

第２１回：Filter map

val numbers = setOf(1, 2, 3)
println(numbers.map { it * 3 }) // [3, 6, 9]

Max min

第２７回：Max min

listOf(1, 42, 4).maxOrNull() == 42
listOf("a", "ab").minByOrNull(String::length) == "a"

解答例

fun doSomethingWithCollection(collection: Collection<String>): Collection<String>? {

    val groupsByLength = collection.groupBy { s -> s.length }

    val maximumSizeOfGroup = groupsByLength.values.map { group -> group.size }.maxOrNull()

    return groupsByLength.values.firstOrNull { group -> group.size == maximumSizeOfGroup }
}