二叉树的下一个结点

最新推荐文章于 2021-07-20 14:42:37 发布

Waitt_

最新推荐文章于 2021-07-20 14:42:37 发布

阅读量346

点赞数

文章标签：剑指OFFER C++

本文链接：https://blog.csdn.net/Waitt_/article/details/85217665

版权

剑指OFFER题25------按牛客网通过率排序

时间：2018.12.22.2019
作者：Waitt

题目

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

时间限制：1秒空间限制：32768K 热度指数：116588

解答

在这里插入图片描述
如图所示，可分为两类：

有右子树的，那么下个结点就是右子树最左边的点；（eg：D，B，E，A，C，G）
没有右子树的，也可以分成两类：
-2.1 是父节点左孩子（eg：N，I，L），那么父节点就是下一个节点；
-2.2 是父节点的右孩子（eg：H，J，K，M）找他的父节点的父节点的父节点…直到当前结点是其父节点的左孩子位置。如果没有eg：M，那么他就是尾节点。

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/
class Solution {
public:
    TreeLinkNode* zhon(TreeLinkNode* pNode)//有右孩子时的函数，可用循环写进主函数中
    {
        if(!pNode->left)
            return pNode;
        return zhon(pNode->left);
    }
    
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if (!pNode)
            return pNode;
        if(pNode->right)//当前结点有右孩子时。
            return zhon(pNode->right);
        TreeLinkNode* fu=pNode->next;//当前结点的父结点
        while(fu)//当前结点无右孩子，则应该寻找其父结点
        {
            if(pNode==(fu->left))//当前结点为父结点的左孩子，则父结点为下一结点
                return fu;
                
            //当前结点为父结点的右孩子，则继续寻找父结点
            pNode=fu;
            fu=fu->next;
        }//根节点的父结点为null，说明此结点为最右结点
        return fu;
    }
};