Question

我想为二进制树实现级别功能，该功能将二进制树转换为列表列表int，每个列表int需要包含当前级别的所有值：例如：

              9
         6          10
     5     7              11
 42   
 42
 42

[[9]; [6; 10]; [5; 7; 11]; [42]; [42]; [42]]

我需要使用这种类型和fold_bin_tree

type 'a bin_tree =
   Node of 'a bin_tree * 'a * 'a bin_tree | Null;;


let rec fold_bin_tree f a t =
  match t with
  Null ->a |
  Node (l,x,r) ->f x (fold_bin_tree f a l) (fold_bin_tree f a r);;

我已经实现了合并功能作为助手，这就是我的代码：

let merge l1 l2 = (*[[a b c] [d e]] merge [[1] [2 3] [4]] = 
[[a b c 1] [d e 2 3] [4]]*)
let rec scan l1 l2 acc =
  match l1,l2 with
  | [],[] -> acc
  | h1::t1,[] -> scan t1 l2 (h1::acc)
  | [], h2::t2 -> scan l1 t2 (h2::acc)
  | h1::t1,h2::t2 -> scan t1 t2 ((h1@h2)::acc)
  in scan l1 l2 [];;

let levels t =
  fold_bin_tree
  (fun x levels_l levels_r -> let lev = merge levels_l levels_r in
    [x]::lev
  ) [] t;;

我已经测试过合并，它似乎运行良好，但是由于

，我的级别无法正常工作

levels example

返回[[9]; [5; 7]; [42]; [42]; [42; 11]; [6; 10]]

其中一个例子是我在帖子开头显示的那棵树

  let example = Node(  Node(Node(Node(Node(Node(Null,42,Null),42,Null),42,Null),5,Null), 6 ,Node( Null, 7 ,Null )) ,9, (Node(Null, 10 ,Node(Null,11,Null))));;

Answer 1

提示：函数merge为什么返回

 merge [[1];[2]] [];;

[[2]; [1]]

？

Ocaml，二叉树的级别功能

1 个答案: