题解 | #二叉树的下一个结点#

二叉树的下一个结点

http://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e

暴力解法当然就是按照中序遍历一遍,但这题中参数只给了待查找的子树根节点,中序遍历操作过于麻烦,最好是从子树根节点直接开始查找。

  1. 如果子树根节点存在右节点,则从该右节点开始,一路向左查找;

  2. 否则,子树根节点的中序遍历下一个节点只能在其父辈中查找,并且只有将当前根节点作为左节点的父节点符合条件:

    1. 往上查找,找到第一个把当前根节点当做左子节点的父节点;

    2. 如果不存在,则待查找子树根节点是该树中序遍历的最后一个节点,返回NULL。

class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode) {
        if(!pNode)
            return NULL;
        TreeLinkNode* temp = NULL;
        if(pNode->right){
            temp = pNode->right;
            while(temp->left){
                temp = temp->left;
            }
        }
        else if(pNode->next){
            while(pNode->next && pNode->next->right == pNode){
                pNode = pNode->next;
            }
            temp = pNode->next;
        }
        return temp;
    }
};
全部评论

相关推荐

06-16 15:04
黑龙江大学 Java
零OFFER战士:另一个版本查看图片
点赞 评论 收藏
分享
这算盘打的
程序员小白条:都这样的,都是潜规则,你自己说可以实习一年就行了,实习可以随便跑路的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务