tp8 step 2 solution

4sh · Feb 14, 2025 · 6020c7f · 6020c7f
1 parent 327506f
commit 6020c7f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tp8/src/main/kotlin/fmt/kotlin/fundamentals/Cellar.kt b/tp8/src/main/kotlin/fmt/kotlin/fundamentals/Cellar.kt
@@ -4,7 +4,9 @@ data class Vineyard(
     val cellars: List<Cellar>
 ) {
 
-    fun countBottles() = emptyMap<String, Int>()
+    fun countBottles() = cellars.fold(emptyMap<String, Int>()) { res, cellar ->
+        res.merge(cellar.bottles.groupingBy { it.name }.eachCount()) { n, m -> n + m }
+    }
 }
 
 data class Cellar(