设计一个算法,找出二叉查找树中指定结点的“下一个“结点(也即中序后继)。可以假定每个结点都含有指向父结点的连接。
设计一个算法,找出二叉查找树中指定结点的“下一个“结点(也即中序后继)。可以假定每个结点都含有指向父结点的连接。
一贱书生 发表于1年前
设计一个算法,找出二叉查找树中指定结点的“下一个“结点(也即中序后继)。可以假定每个结点都含有指向父结点的连接。
  • 发表于 1年前
  • 阅读 23
  • 收藏 0
  • 点赞 0
  • 评论 0

新睿云服务器60天免费使用,快来体验!>>>   

问题分析

假设需要查找 node 结点的下一个结点,需要考虑三种情况:

①node 节点有右孩子

下一个结点就是以node结点的右孩子为根的子树中的最左下结点。如下图:node=8,它的下一个结点为12.

 

②node 节点没有右孩子时,node节点是其父结点的左孩子。如下图,结点8的下一个结点是结点12

 

 

③node 节点没有右孩子时,node节点是其父结点的右孩子,如下图,结点14 的下一个结点是 结点16

 

 

 

如何在一棵二叉树中查找某个结点?

可以用先序遍历的思路。但是一定要注意递归的实现---第11行的 return target 是很有意义的。

在每一次的递归调用中,每个方法都有一个自己的 target 局部变量。如果在这个方法里面找到了目标结点,如果没有 return 返回的话,当递归回退时就会丢失“已找到的目标结点”----即上一层的 find 方法中的target 变量还是null(尽管在下一层递归中已经找到了目标结点)

通过第11行的 return 语句,如果在某一层还未找到目标结点,则会继续递归调用下去,如果找到了,在 return 的时候,上一层的find方法的target变量就不会为 null ,  从而在最终 find 方法结束时,返回找到的目标结点。

 

复制代码

 1     //采用先序遍历查找值为ele的结点
 2     private BinaryNode find(BinaryNode root, int ele){
 3         if(root == null)
 4             return null;
 5         if(root.ele == ele)
 6             return root;
 7         BinaryNode target = null;
 8         target = find(root.left, ele);
 9         if(target == null)//如果左子树中没有值为ele的结点,if成立,在右子树中查找
10             target = find(root.right, ele);
11         return target;
12     }

复制代码

①第3-4行是应对查找到叶子结点的情况

②第5-6行,是成功查找到了指定结点的情况。(①②类似于先序遍历中的访问根结点)

③第8行,表示先在左子树中查找(类似于先序遍历左子树)

④第9-10行if表示在左子树中未查找到该结点,则查找右子树⑤第11行,返回查找的结点。若返回null,表示未找到

 

三,代码分析

①node 节点有右孩子

复制代码

1 if(node.right != null)
2 {
3     BinaryNode current = node.right;
4     while(current.left != null)
5     {
6     current = current.left;
7     }
8     nextNode = current;
9  }

复制代码

第3行,先定位到node结点的右孩子。

第4行while循环,查找最左下结点

 

②node 节点没有右孩子时,node节点是其父结点的左孩子

1 else if(node.parent != null){//node结点 是 parent 的孩子
2     if(node.parent.left != null && node.parent.left.equals(node))// node 是 parent 的左孩子
3         nextNode = node.parent;

如果node节点是其父结点的左孩子,那么下一个结点就是node节点的父结点。

 

③node 节点没有右孩子时,node节点是其父结点的右孩子

复制代码

1 else{//node 是 parent的右孩子
2     BinaryNode current = node.parent;
3     //一直往着右孩子的父结点指针向上走
4     while(current.parent.right != null && current.parent.right.equals(current))
5     {
6         current = current.parent;
7     }
8     nextNode = current.parent;
9 }

复制代码

要注意第4行while循环中的第一个条件:current.parent.right != null

为什么不要判断 current.parent != null 呢?因为在前面的if语句中已经判断了(if(node.parent != null)

 

完整代码:

public class BSTNextNode {

    private class BinaryNode{
        int ele;
        BinaryNode left;
        BinaryNode right;
        BinaryNode parent;
        int hash;//cache hashCode
        
        public BinaryNode(int ele) {
            this.ele = ele;
            parent = left = right = null;
        }
        
        @Override
        public boolean equals(Object obj) {
            if(obj == null)
                return false;
            if(!(obj instanceof BinaryNode))
                return false;
            BinaryNode node = (BinaryNode)obj;
            return node.ele == this.ele;
        }
        
        @Override
        public int hashCode() {// 参考《effective java》中覆盖equals方法
            int result = hash;
            if(result == 0){
                result = 17;
                result = result*31 + ele;
                hash = result;
            }
            return result;
        }

或者:

 

下面是该算法的实现代码(已正确处理结点为空的情况)

public TreeNode inorderSucc(TreeNode n)
{
if(n==null) return null;
/*
找到右子结点,则返回右子树里最左边的结点
*/
if(n.right!=null)
{
return leftMostChild(n.right);
}
else
{
TreeNode q=n;
TreeNode x=q.parent;
//向上直至位于左边而不是右边
while(x!=null&&x.left!=q)
{
q=x;
x=x.parent;
}
return x;
}
}


public TreeNode leftMostChild(TreeNode n)
{
if(n==null)
return null;
while(n.left!=null)
{
n=n.left;
}
return null;

}

 

 

或者

 

 

  • 打赏
  • 点赞
  • 收藏
  • 分享
共有 人打赏支持
粉丝 17
博文 722
码字总数 600072
×
一贱书生
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: