58、二叉树的下一个节点

最新推荐文章于 2022-04-18 14:08:10 发布

原创最新推荐文章于 2022-04-18 14:08:10 发布 · 172 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer

剑指offer-树专栏收录该内容

14 篇文章

订阅专栏

博客围绕二叉树展开，给定一个含指向父结点指针的二叉树及其中一个结点，要找出中序遍历顺序的下一个结点。给出两种思路，一是若节点右孩子存在，从右孩子出发沿左子结点指针找叶子节点；二是若节点非根节点，是父节点左孩子则返回父节点，否则向上遍历。

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

思路：若节点右孩子存在，则设置一个指针从该节点的右孩子出发，一直沿着指向左子结点的指针找到的叶子
节点即为下一个节点；若节点不是根节点。如果该节点是其父节点的左孩子，则返回父节点；否则继续向上遍
历其父节点的父节点，重复之前的判断，返回结果

public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode node)
    {
       if(node==null)
           return null;
        if(node.right != null)//如果有右子树，则找右子树的最左节点
        {
            node = node.right;
            while(node.left != null)
                node = node.left;
            return node;
           
        }
        
        while(node.next!=null)//没右子树，则找第一个当前节点是父节点左孩子的节点
        {
           if (node.next.left==node)
               return node.next;
            node=node.next;
        }

        return null;   //退到了根节点仍没找到，则返回null
        
    }
}