剑指offer（57）二叉树的下一个节点

Posted 2020-12-31 Blog of Eric Wu

题目描述

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

 

题目分析

这题一定要画图，因为只有画图我们才能分清楚下一个节点有哪些情况

1.如果一个节点有右子树，那么它的下一个节点就是它的右子树中的最左子节点。也就是说，从右子节点出发一直沿着指向左子节点的指针，我们就能找到下一个节点。

2.如果没有右子树，又可以分为两种情况

• 如果节点是它父节点的左子节点，那么它的下一个节点就是它的父节点。
• 如果一个节点既没有右子树，并且它还是父节点的右子节点，那么需要沿着指向父节点的指针一直向上便利，直到找到一个是它父节点的左子节点的节点。

 

代码

function GetNext(pNode)
{
    if(pNode == null){
        return null;
    }
    if(pNode.right != null){//第1种
        pNode = pNode.right;
        while(pNode.left != null){
            pNode = pNode.left;
        }
        return pNode;
    }
    while(pNode.next != null){ //第2种
        if(pNode == pNode.next.left){ 
            return pNode.next;
        }
        pNode = pNode.next;
    }
    return null;
}