scala--简洁优雅的树定义

也许还不够优雅。

 

trait Tree[+T] {
   /**
    * 深度优先遍历
    */
   def dfs(func: T => Unit) {
      this match {
         case Empty                    =>
         case Node(value, left, right) => func(value); left.dfs(func); right.dfs(func)
      }
   }
}
/**
 * 空树
 */
case object Empty extends Tree[Nothing]()
/**
 * 节点, 单个节点是一棵树
 */
case class Node[T](val value: T, val left: Tree[T], val right: Tree[T]) extends Tree[T]
/**
 * 叶子，只是为了创建更方便
 */
case class Leaf[T](override val value: T) extends Node[T](value, Empty, Empty)

 在类型上对树做递归定义，函数型语言显得更优雅。

• 空树是一棵树
• 节点及其左右子树构成一棵树

关于叶子节点不是必要的，只是为了创建时简洁

 

看这个例子

 val tree = Node(1,
         Node(2, Leaf(4), Empty),
         Node(3, Leaf(5), Empty))

 

构建了形如

 的一颗树，并对其进行深度优先遍历