剑指offer JS题解 (57)二叉树的下一个结点

题目描述

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

解题思路

leetcode上好像没有相关的题目，是在牛客上做的。首先要看一下树结点的结构，是具备左右结点的指针以及指向父节点的指针（next结点）的。这就使得我们在中序遍历时可以追溯二叉树要遍历的下一个结点。

那我们中序遍历的时候是怎么做的？左-中-右。现在给到一个结点pNode，它可以是在左中右这三个位置上，所以需要具体讨论。

首先第一个情况：结点作为根节点，此时它有右子树的话，那下一个结点就是右子树的最左叶结点了。

第二个情况：假如结点没有右子树，那它就得往父结点上面找下一个结点了。
有两种情况可以具体分析一下（可以画画图），首先它可能是作为某个结点的一个右结点，并拥有一颗左子树，这时它的下一个结点是哪里呢？它的父结点显然已经被遍历过了，所以下一个结点应该是它的父节点的某个祖宗结点（也就是向上追溯，直到有一个父结点是作为树的左节点）。

另一种情况，它可能恰好就是某颗树的左节点，那它的下一个结点就是它的父亲，这时候就满足pNode===pNode.next.left，可以直接返回父节点了。

Code

/*function TreeLinkNode(x){
    this.val = x;
    this.left = null;
    this.right = null;
    this.next = null;
}*/
function GetNext(pNode)
{
    // write code here
    // 中序遍历，首先获取到根节点，接下来就要去判断右子树
    if(!pNode) return null;
    // 情况一：右子树非空，取右子树的最左子节点
    if(pNode.right){
        pNode=pNode.right;
        while(pNode.left){
            pNode=pNode.left;
        }
        return pNode;
    }
    // 情况二：右子树空，判断当前节点是否是父节点的左节点
    // 是的话就返回它的父节点作为下一个结点
    while(pNode.next){
        if(pNode===pNode.next.left){
            return pNode.next;
        }
        // 不满足，则通过next回溯父节点，直到有一个结点等于其父节点的左节点
        pNode=pNode.next;
    }
    return null;
}

运行环境：JavaScript (V8 6.0.0)
运行时间：14ms
占用内存：11692k