Source file wtree.ml

(* This file is part of asak.
 *
 * Copyright (C) 2019 IRIF / OCaml Software Foundation.
 *
 * asak is distributed under the terms of the MIT license. See the
 * included LICENSE file for details. *)

type 'a wtree =
  | Node of (int * 'a wtree * 'a wtree)
  | Leaf of 'a

let fold_tree n l =
  let rec aux = function
    | Leaf a -> l a
    | Node (f,a,b) -> n f (aux a) (aux b)
  in aux

let size_of_tree f =
  fold_tree (fun _ a b -> 1 + a + b) f