二叉树的下一个结点

最新推荐文章于 2023-12-02 18:51:32 发布

原创最新推荐文章于 2023-12-02 18:51:32 发布 · 156 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种高效算法，用于在给定二叉树及其一个节点的情况下，找出该节点在中序遍历顺序下的下一个节点。算法区分了节点有无右子树的情况，对于有右子树的节点，下一个节点为其右子树中最左侧的节点；对于无右子树的节点，算法会向上查找直至找到当前节点为其父节点左子树的情况。

题目：

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

解法：

图片来源：牛客网
如图所示，可以归纳出两种情况：1）给定节点有右儿子，则下一个节点是右儿子的最左边儿子，2）给点节点没有右儿子，则一直找到当前节点的父节点是其父节点的左儿子的节点，比如（j,H 节点）。

代码如下：

class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}

public class GetNext {

    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        if(pNode == null) {
            return pNode;
        }
        if(pNode.right != null){
            pNode = pNode.right;
            while (pNode.left!=null){
                pNode = pNode.left;
            }
            return pNode;
        }
        while (pNode.next != null){
            if(pNode.next.left == pNode){
                return pNode.next;
            }
            pNode = pNode.next;
        }

        return null;

    }
}