本文的代码来自于《数据结构与算法(JAVA语言版)》,是笔者在网上找到的资料,非正式出刊版物。笔者对代码一些比较难以理解的部分添加了注释和图解,欢迎大家来讨论。
二叉树查找的基本思想是在二叉查找树中从根节点开始,如果大于根节点则继续比较右孩子,如果小于则继续查找左孩子,依次往复。
如图所示
输入:待查元素ele
输出:对应元素在二叉查找树中的结点位置
代码:
public Node search(Object ele){
return binTSearchRe (root, ele);
}
private Node binTSearchRe(BinTreeNode rt, Object ele){
if (rt==null) return null;
switch(strategy.compare(ele,rt.getData())){
case 0: return rt; //等于
case -1: return binTSearchRe(rt.getLChild(),ele); //小于
default: return binTSearchRe(rt.getRChild(),ele); //大于
}
}
输入:待查元素ele
输出:对应元素在二叉查找树中的结点位置
代码:
public Node search(Object ele){
return binTSearchRe (root, ele);
}
private Node binTSearchRe(BinTreeNode rt, Object ele){
if (rt==null) return null;
switch(strategy.compare(ele,rt.getData())){
case 0: return rt; //等于
case -1: return binTSearchRe(rt.getLChild(),ele); //小于
default: return binTSearchRe(rt.getRChild(),ele); //大于
}
}
输入:根结点v
输出:在v 为根的二叉查找树中最小元素的位置
代码:
public Node min(BinTreeNode v){
if (v!=null)
while (v.hasLChild()) v = v.getLChild();
return v;
}
输入:根结点v
输出:在v 为根的二叉查找树中最大元素的位置
代码:
public Node max(BinTreeNode v){
if (v!=null)
while (v.hasRChild()) v = v.getRChild();
return v;
}
输入:根结点v
输出:返回v 在中序遍历序列中的后续结点
代码:
private BinTreeNode getSuccessor (BinTreeNode v){
if (v==null) return null;
if (v.hasRChild()) return (BinTreeNode)min(v.getRChild());
while (v.isRChild()) v = v.getParent();
return v.getParent();
}
输入:根结点v
输出:返回v 在中序遍历序列中的前驱结点
代码:
private BinTreeNode getPredecessor(BinTreeNode v){
if (v==null) return null;
if (v.hasLChild()) return (BinTreeNode)max(v.getLChild());
while (v.isLChild()) v = v.getParent();
return v.getParent();
}