···
import gleam/dict.{type Dict}
import gleam/dynamic/decode
import gleam/int.{to_string}
import gleam/io.{println}
import gleam/result.{unwrap}
-
import gleam/string.{trim}
···
-
fn get_total_number_dict(data: Dict(String, Nested)) {
-
|> dict.fold(0, fn(acc, _, data) {
-
NestedList(items) -> get_total_number_list(items)
-
NestedDict(items) -> get_total_number_dict(items)
-
fn get_total_number_list(data: List(Nested)) {
-
|> list.fold(0, fn(acc, data) {
-
NestedList(items) -> get_total_number_list(items)
-
NestedDict(items) -> get_total_number_dict(items)
-
fn get_total_number_dict_no_red(data: Dict(String, Nested)) {
-
|> dict.fold(False, fn(state, _, data) {
-
StringValue("red") -> True
-
|> dict.fold(0, fn(acc, _, data) {
-
NestedList(items) -> get_total_number_list_no_red(items)
-
NestedDict(items) -> get_total_number_dict_no_red(items)
-
fn get_total_number_list_no_red(data: List(Nested)) {
-
|> list.fold(0, fn(acc, data) {
-
NestedList(items) -> get_total_number_list_no_red(items)
-
NestedDict(items) -> get_total_number_dict_no_red(items)
···
|> json.parse(int_anywhere_decoder())
-
get_total_number_list([input])
-
get_total_number_list_no_red([input])
···
import gleam/dict.{type Dict}
import gleam/dynamic/decode
import gleam/int.{to_string}
import gleam/io.{println}
import gleam/result.{unwrap}
···
+
fn handle_calc(data: Nested, no_red) {
+
NestedList(items) -> get_total_number_list(items, no_red)
+
NestedDict(items) -> get_total_number_dict(items, no_red)
+
fn get_total_number_dict(data: Dict(String, Nested), no_red) {
+
let has_red = case no_red {
+
|> dict.fold(False, fn(state, _, data) {
+
StringValue("red") -> True
+
|> dict.fold(0, fn(acc, _, data) { acc + handle_calc(data, no_red) })
+
fn get_total_number_list(data: List(Nested), no_red no_red) {
+
|> list.fold(0, fn(acc, data) { acc + handle_calc(data, no_red) })
···
|> json.parse(int_anywhere_decoder())
+
get_total_number_list([input], no_red: False)
+
get_total_number_list([input], no_red: True)