commit ad19977ae0d70d52e9a85f13788e4327f3347fee · aylac.top/advent_of_code

+27 -30
2015/12/gleam/src/main.gleam
···

       23
       23
        
         ])

     

       24
       24
        
       }

     

       25
       25
        
       

     

       26
       26
       -
       fn handle_calc(data: Nested, no_red) {

     

       26
       26
       +
       fn get_total_number(data: Nested, no_red no_red) {

     

       27
       27
        
         case data {

     

       28
       28
       -
           NestedList(items) -> get_total_number_list(items, no_red)

     

       29
       29
       -
           NestedDict(items) -> get_total_number_dict(items, no_red)

     

       28
       28
       +
           NestedList(items) ->

     

       29
       29
       +
             items

     

       30
       30
       +
             |> list.fold(0, fn(acc, data) { acc + get_total_number(data, no_red) })

     

       31
       31
       +
           NestedDict(items) -> {

     

       32
       32
       +
             let has_red = case no_red {

     

       33
       33
       +
               True ->

     

       34
       34
       +
                 items

     

       35
       35
       +
                 |> dict.fold(False, fn(state, _, data) {

     

       36
       36
       +
                   case data {

     

       37
       37
       +
                     StringValue("red") -> True

     

       38
       38
       +
                     _ -> state

     

       39
       39
       +
                   }

     

       40
       40
       +
                 })

     

       41
       41
       +
               False -> False

     

       42
       42
       +
             }

     

       43
       43
       +
             case has_red {

     

       44
       44
       +
               True -> 0

     

       45
       45
       +
               False ->

     

       46
       46
       +
                 items

     

       47
       47
       +
                 |> dict.fold(0, fn(acc, _, data) {

     

       48
       48
       +
                   acc + get_total_number(data, no_red)

     

       49
       49
       +
                 })

     

       50
       50
       +
             }

     

       51
       51
       +
           }

     

       30
       52
        
           IntValue(v) -> v

     

       31
       53
        
           _ -> 0

     

       32
       54
        
         }

     

       33
       55
        
       }

     

       34
       56
        
       

     

       35
       35
       -
       fn get_total_number_dict(data: Dict(String, Nested), no_red) {

     

       36
       36
       -
         let has_red = case no_red {

     

       37
       37
       -
           True ->

     

       38
       38
       -
             data

     

       39
       39
       -
             |> dict.fold(False, fn(state, _, data) {

     

       40
       40
       -
               case data {

     

       41
       41
       -
                 StringValue("red") -> True

     

       42
       42
       -
                 _ -> state

     

       43
       43
       -
               }

     

       44
       44
       -
             })

     

       45
       45
       -
           False -> False

     

       46
       46
       -
         }

     

       47
       47
       -
         case has_red {

     

       48
       48
       -
           True -> 0

     

       49
       49
       -
           False ->

     

       50
       50
       -
             data

     

       51
       51
       -
             |> dict.fold(0, fn(acc, _, data) { acc + handle_calc(data, no_red) })

     

       52
       52
       -
         }

     

       53
       53
       -
       }

     

       54
       54
       -
       

     

       55
       55
       -
       fn get_total_number_list(data: List(Nested), no_red no_red) {

     

       56
       56
       -
         data

     

       57
       57
       -
         |> list.fold(0, fn(acc, data) { acc + handle_calc(data, no_red) })

     

       58
       58
       -
       }

     

       59
       59
       -
       

     

       60
       57
        
       pub fn main() {

     

       61
       58
        
         let assert Ok(input) =

     

       62
       59
        
           read(from: "../input.json")

     
···

       64
       61
        
           |> json.parse(int_anywhere_decoder())

     

       65
       62
        
       

     

       66
       63
        
         println(

     

       67
       67
       -
           get_total_number_list([input], no_red: False)

     

       64
       64
       +
           get_total_number(input, no_red: False)

     

       68
       65
        
           |> to_string,

     

       69
       66
        
         )

     

       70
       67
        
         println(

     

       71
       71
       -
           get_total_number_list([input], no_red: True)

     

       68
       68
       +
           get_total_number(input, no_red: True)

     

       72
       69
        
           |> to_string,

     

       73
       70
        
         )

     

       74
       71
        
       }