Implement efficient algorithm in TF

2024/bonus/day11/step/main.tf

@@ -1,14 +1,21 @@
 variable "prev" {
-  type = list(number)
+  type = map(number)
+}
+
+module "transform" {
+  source   = "../transform"
+  for_each = var.prev
+
+  num = each.key
 }
 
 locals {
-  values = [
-    for num in var.prev : num == 0 ? [1]
-    : length(tostring(num)) % 2 == 0
-    ? [tonumber(substr(tostring(num), 0, length(tostring(num)) / 2)), tonumber(substr(tostring(num), length(tostring(num)) / 2, 10))]
-    : [num * 2024]
-  ]
+  by_value = flatten([
+    for key, value in module.transform :
+    [for result in value.result : { num = result, count = var.prev[key] }]
+  ])
+
+  grouped = { for kv in local.by_value : kv.num => kv.count... }
 }
 
 # module "transform" {
@@ -23,5 +30,5 @@ locals {
 # }
 
 output "next" {
-  value = flatten(local.values)
+  value = { for num, groups in local.grouped : num => sum(groups) }
 }