2014-06-25 60 views
-1

这是鉴于这样的二进制搜索和布局它OCaml 99布局二叉搜索树

问题65:

enter image description here


的y轴节点很容易,因为它只是级别编号,从1开始。

节点的x asix稍微复杂一点,但通过观察,假设整个树的高度为h,那么节点的x就是左边的最大大小,就好像它是一棵满树, x = 2^(h-y)-1

但是,还有一个特殊情况,其中最左边的节点的x始终是我们需要处理的。

这里是我的代码:

type 'a btree = Empty | Node of 'a * 'a btree * 'a btree 

type 'a pos_binary_tree = 
    | E (* represents the empty tree *) 
    | N of 'a * int * int * 'a pos_binary_tree * 'a pos_binary_tree 

let rec height = function 
    | Empty -> 0 
    | Node (_,l,r) -> 1 + max (height l) (height r) 

let get_fullsize h level = (2. ** (Float.of_int (h+1-level)) |> Int.of_float) - 1 

let layout_btree2_correct t = 
    let h = height t in 
    let rec lay off y = function 
    | Empty -> get_fullsize h y, E 
    | Node (w, Empty, r) when off = 0 -> 
     let wtr, newr = lay 1 (y+1) r in 
     1+wtr, N (w, 1, y+1, E, newr) 
    | Node (w, l, r) -> 
     let wt1, newl = lay off (y+1) l in 
     let wt2, newr = lay (off+wt1+1) (y+1) r in 
     wt1+wt2+1, N (w, off+wt1+1, y, newl, newr) 
    in 
    lay 0 1 t |> snd 

我确实是:

  1. 得到整个树的高度
  2. 总是返回回全宽,它可能占用
  3. x应该是左节点宽度+ 1
  4. 对于最左边的节点的特殊情况下,返回1个+宽的权利,宽度

在我的方式,我必须先前往树一次得到的高度。任何人都可以建议更好的实施,例如,只是一次旅行树?

回答