二叉树的下一个结点

最新推荐文章于 2023-11-06 00:22:49 发布

_yxy_

最新推荐文章于 2023-11-06 00:22:49 发布

阅读量183

点赞数

分类专栏： C与C++ 剑指offer 文章标签：剑指offer C++ 二叉树

本文链接：https://blog.csdn.net/u011643312/article/details/72991989

版权

C与C++ 同时被 2 个专栏收录

64 篇文章 0 订阅

订阅专栏

剑指offer

42 篇文章 0 订阅

订阅专栏

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

主要是当前结点如果是父节点的右孩子时的处理：需要一直上溯，找到是左孩子的结点

/*
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!=NULL){
            pNode=pNode->right;
            while(pNode->left!=NULL)
                pNode=pNode->left; 
            return pNode;
        }
        else if(pNode->next==NULL)
            return NULL;
        else if(pNode->next->left==pNode)
            return pNode->next;
        else{
            while(pNode->next->next!=NULL&&pNode->next->next->right==pNode->next)
                pNode=pNode->next;
            return pNode->next->next;
        }
    }
};