二叉树的下一个结点

最新推荐文章于 2023-08-22 22:35:56 发布

genius-x

最新推荐文章于 2023-08-22 22:35:56 发布

阅读量127

点赞数

本文链接：https://blog.csdn.net/qq_25327609/article/details/82860869

版权

算法同时被 2 个专栏收录

58 篇文章 0 订阅

订阅专栏

c++

50 篇文章 1 订阅

订阅专栏

题目描述：

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

代码

/*
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* GetNext(TreeLinkNode* pNode)
    {
    	//是否有右孩子
        if(pNode->right!=nullptr)
        {
            return FindFirst(pNode->right);       
        }
        //无右孩子，在父结点做文章
        else
        {
        	//父结点
           TreeLinkNode* father=pNode->next;
            if(father==nullptr)
                return nullptr;
             //是父结点的左孩子 即为下一个访问节点
            if(father->left==pNode)
                return father;
                //是父结点的右孩子 往父节点回溯 直到找到下一个当前子树属于祖先节点的左子树
            else
            {
                TreeLinkNode* temp=father->next;
                while(temp&&temp->right==father)
                {
                    father=temp;
                    temp=temp->next;
                }
                //可能找到  也可能为空（当前节点为最后一个访问节点）
                return temp;
            }
            
        }
        
    }
    //找中序遍历第一个值
    TreeLinkNode* FindFirst(TreeLinkNode* root)
    {
        if(root->left==nullptr)
            return root;
        else
            return FindFirst(root->left);
    }
};