14 lines
382 B
Text
14 lines
382 B
Text
@[reducible] def Map : Type := RBMap Nat Bool (<)
|
|
|
|
def mkMapAux : Nat → Map → Map
|
|
| 0 m := m
|
|
| (n+1) m := mkMapAux n (m.insert n (n % 10 = 0))
|
|
|
|
def mkMap (n : Nat) :=
|
|
mkMapAux n {}
|
|
|
|
def main (xs : List String) : IO UInt32 :=
|
|
let m := mkMap xs.head.toNat in
|
|
let v := RBMap.fold (λ (k : Nat) (v : Bool) (r : Nat), if v then r + 1 else r) m 0 in
|
|
IO.println (toString v) *>
|
|
pure 0
|