commit d3782cd663a2aa4047edc3ad4833d9050418d264 · aylac.top/advent_of_code

+24 -56
2015/12/gleam/src/main.gleam
···

       1
       1
        
       import gleam/dict.{type Dict}

     

       2
       2
       -
       import gleam/dynamic

     

       3
       2
        
       import gleam/dynamic/decode

     

       4
       3
        
       import gleam/int.{to_string}

     

       5
       4
        
       import gleam/io.{println}

     

       6
       5
        
       import gleam/json

     

       7
       6
        
       import gleam/list

     

       8
       7
        
       import gleam/result.{unwrap}

     

       9
       9
       -
       import gleam/string.{trim}

     

       10
       8
        
       import simplifile.{read}

     

       11
       9
        
       

     

       12
       10
        
       type Nested {

     
···

       25
       23
        
         ])

     

       26
       24
        
       }

     

       27
       25
        
       

     

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

     

       29
       29
       -
         data

     

       30
       30
       -
         |> dict.fold(0, fn(acc, _, data) {

     

       31
       31
       -
           let add = case data {

     

       32
       32
       -
             NestedList(items) -> get_total_number_list(items)

     

       33
       33
       -
             NestedDict(items) -> get_total_number_dict(items)

     

       34
       34
       -
             IntValue(v) -> v

     

       35
       35
       -
             _ -> 0

     

       36
       36
       -
           }

     

       37
       37
       -
           acc + add

     

       38
       38
       -
         })

     

       39
       39
       -
       }

     

       40
       40
       -
       

     

       41
       41
       -
       fn get_total_number_list(data: List(Nested)) {

     

       42
       42
       -
         data

     

       43
       43
       -
         |> list.fold(0, fn(acc, data) {

     

       44
       44
       -
           let add = case data {

     

       45
       45
       -
             NestedList(items) -> get_total_number_list(items)

     

       46
       46
       -
             NestedDict(items) -> get_total_number_dict(items)

     

       47
       47
       -
             IntValue(v) -> v

     

       48
       48
       -
             _ -> 0

     

       49
       49
       -
           }

     

       50
       50
       -
           acc + add

     

       51
       51
       -
         })

     

       26
       26
       +
       fn handle_calc(data: Nested, 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)

     

       30
       30
       +
           IntValue(v) -> v

     

       31
       31
       +
           _ -> 0

     

       32
       32
       +
         }

     

       52
       33
        
       }

     

       53
       34
        
       

     

       54
       54
       -
       fn get_total_number_dict_no_red(data: Dict(String, Nested)) {

     

       55
       55
       -
         let has_red =

     

       56
       56
       -
           data

     

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

     

       58
       58
       -
             case data {

     

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

     

       60
       60
       -
               _ -> state

     

       61
       61
       -
             }

     

       62
       62
       -
           })

     

       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
       +
         }

     

       63
       47
        
         case has_red {

     

       64
       48
        
           True -> 0

     

       65
       49
        
           False ->

     

       66
       50
        
             data

     

       67
       67
       -
             |> dict.fold(0, fn(acc, _, data) {

     

       68
       68
       -
               let add = case data {

     

       69
       69
       -
                 NestedList(items) -> get_total_number_list_no_red(items)

     

       70
       70
       -
                 NestedDict(items) -> get_total_number_dict_no_red(items)

     

       71
       71
       -
                 IntValue(v) -> v

     

       72
       72
       -
                 _ -> 0

     

       73
       73
       -
               }

     

       74
       74
       -
               acc + add

     

       75
       75
       -
             })

     

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

     

       76
       52
        
         }

     

       77
       53
        
       }

     

       78
       54
        
       

     

       79
       79
       -
       fn get_total_number_list_no_red(data: List(Nested)) {

     

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

     

       80
       56
        
         data

     

       81
       81
       -
         |> list.fold(0, fn(acc, data) {

     

       82
       82
       -
           let add = case data {

     

       83
       83
       -
             NestedList(items) -> get_total_number_list_no_red(items)

     

       84
       84
       -
             NestedDict(items) -> get_total_number_dict_no_red(items)

     

       85
       85
       -
             IntValue(v) -> v

     

       86
       86
       -
             _ -> 0

     

       87
       87
       -
           }

     

       88
       88
       -
           acc + add

     

       89
       89
       -
         })

     

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

     

       90
       58
        
       }

     

       91
       59
        
       

     

       92
       60
        
       pub fn main() {

     
···

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

     

       97
       65
        
       

     

       98
       66
        
         println(

     

       99
       99
       -
           get_total_number_list([input])

     

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

     

       100
       68
        
           |> to_string,

     

       101
       69
        
         )

     

       102
       70
        
         println(

     

       103
       103
       -
           get_total_number_list_no_red([input])

     

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

     

       104
       72
        
           |> to_string,

     

       105
       73
        
         )

     

       106
       74
        
       }