树的前序与后序遍历 (Pre-order Post-order Traversal)

引子

公司里一项目使用work flow, 使用的是JBoss jBPM.存储为xml格式，需求如下

1) xml文档和business model: node class 实例之间的相互转换

2) 检查整个tree的高度和宽度，避免用户创建太深或者太宽的work flow

3) 检查node的名称，不容许有重复的命名

 

xml 样例

<node cap="Root Node" con=""> <node cap="New node1"> <node cap="New node11"> <node cap="New node111"> <node cap="New node1111"> <node cap="New node15"> <node cap="New node16"> <node cap="New node17"> <node cap="New node18"> <node cap="New node19"> <node cap="New node20"> <node cap="New node21"> <node cap="New node22"> <node cap="New node23"> <node cap="New node24"> <node cap="New node25"></node> </node> </node> </node> </node> </node> </node> </node> </node> </node> </node> </node> </node> </node> </node> <node cap="New node2"></node> <node cap="New node3"></node> <node cap="New node4"></node> </node> 

business model

Class Node { Long id; String name; List<Node> childs; } 

Db Design

DB存储时

• 每个WorkFlow存成一条记录
• WorkFlow中的每个节点存成一条记录
• 节点之间的关系通过xml保存在WorkFlow table中

WorkFlow

| id | name | xmlContent|

 

WorkFlowNode

| id | workFlowId | name | 

 

 

 

递归(recursive)

最大高度 (后序，先得到child的高度，然后是当前节点的高度: maxChildHeight + 1)

public int getMaxHeight(Node root) { if(root.childs == null) return 0; int maxChildHeight = -1; for(Node child : root.childs) { int childHeight = getMaxHeight(child); if(maxChildHeight < childHeight) maxChildHeight = childHeight; } return maxChildHeight + 1; } 

 

最大宽度 (前序后序皆可)

public int[] maxWidth(Node root, int[] widthHolder) { if(root.childs.size > widthHolder[0]) widthHolder[0] = root.childs.size for(Node child : root.childs) maxWidth(child, widthHolder) } 

使用数组作为int类型的holder，因为java里面没有指针，而且java的Integer为Immutable类型

非递归(non-recursive) 泛型版

import java.util.List; import java.util.Stack; public abstract class NodeVisitor<T> { public abstract void visit(T cur); public abstract List<T> getChild(T cur); /* * *********************** Non-recursive ************* */ public void visitPreOrder(T root) { Stack<T> statck = new Stack<T>(); statck.push(root); while (!statck.isEmpty()) { T cur = statck.pop(); visit(cur); List<T> childs = getChild(cur); for (T child : childs) { statck.push(child); } } } public void visitPostOrder(T root) { Stack<T> statck = new Stack<T>(); statck.push(root); // need isVisit flag if do not use two stack. Stack<T> statck2 = new Stack<T>(); while (!statck.isEmpty()) { T cur = statck.pop(); statck2.push(cur); List<T> childs = getChild(cur); for (T child : childs) { statck.push(child); } } while (!statck2.isEmpty()) { T cur = statck2.pop(); visit(cur); } } }  

 

• 非递归遍历，使用stack(FILO)。为什么不使用queue(FIFO)？广度优先(Breadth-First)选queue,深度优先(Depth-first)选队列。
• 后序遍历，我们使用两个stack，一个先序stack，一个后序stack，先序statck的出栈顺序即是后序stack的进栈顺序。另外一种方法是使用访问flag。定义一个node wrapper类，增加isVisit属性；在while中使用peek操作获得栈顶元素，如果被访问，则出栈，调用visit方法；没有被访问，则将子节点压栈，标记已访问
• 使用泛型声明，这样这个class被可以访问任意具体Tree结构的class

 

如何传递参数和返回结果？

 

• 使用java inner class 或者 anonymous class 来定义NodeVisitor子类，这样在NodeVisitor的visit方法中可以访问外部closures类的属性

 

使用如下：

int maxChildHeight(Node root) { int[] maxHeightHolder = new int[1]; Map<Node, Int> nodeHeight = new HashMap<Node, Int>(); nodeHeight.put(root, 0); NodeVisitor<Node> visitor = new NodeVisitor<Node>() { public void visit(Node cur) { int h = nodeHeight.get(cur); if(h > maxHeightHolder[0]) maxHeightHolder[o] = h; for(Node child : cur.childs) nodeHeight.put(child, h + 1); } public List<Node> getChild(Node cur) { return cur.childs; } } visitor.visitPreOrder(root); } 

 

需要定义为final类型，否则anonymous class无法访问

 

---

 

java内部突破java思维，几下几点可给力

• 泛型(generic)
• 反射(reflection)
• closures(inner/anonymous class)
• 注解(annotation)