From 0f1e145b80813ae2331b7dac5ace0c589654ad2a Mon Sep 17 00:00:00 2001
From: tchojnacki <tomaszchojnacki2001@gmail.com>
Date: Thu, 11 Aug 2022 19:24:23 +0200
Subject: Move subproject to avoid IntelliJ module name issues

---
 aoc-2021-kotlin/src/Day06.kt | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 aoc-2021-kotlin/src/Day06.kt

(limited to 'aoc-2021-kotlin/src/Day06.kt')

diff --git a/aoc-2021-kotlin/src/Day06.kt b/aoc-2021-kotlin/src/Day06.kt
new file mode 100644
index 0000000..4627cca
--- /dev/null
+++ b/aoc-2021-kotlin/src/Day06.kt
@@ -0,0 +1,38 @@
+object Day06 {
+    private fun calculateFishPopulation(input: String, days: Int): Long {
+        val fishCounts =
+            input
+                .trim()
+                .split(",")
+                .map(String::toInt)
+                .groupingBy { it }
+                .eachCount()
+                .mapValues { (_, v) -> v.toLong() }
+                .toMutableMap()
+
+        repeat(days) {
+            val readyToBirth = fishCounts.getOrDefault(0, 0)
+            repeat(8) {
+                fishCounts[it] = fishCounts.getOrDefault(it + 1, 0)
+            }
+            fishCounts.merge(6, readyToBirth, Long::plus)
+            fishCounts[8] = readyToBirth
+        }
+
+        return fishCounts.values.sum()
+    }
+
+    fun part1(input: String): Int = calculateFishPopulation(input, 80).toInt()
+
+    fun part2(input: String): Long = calculateFishPopulation(input, 256)
+}
+
+fun main() {
+    val testInput = readInputAsString("Day06_test")
+    check(Day06.part1(testInput) == 5934)
+    check(Day06.part2(testInput) == 26984457539)
+
+    val input = readInputAsString("Day06")
+    println(Day06.part1(input))
+    println(Day06.part2(input))
+}
-- 
cgit v1.2.3