找出中序遍历顺序的下一个结点并且返回

最新推荐文章于 2020-08-30 11:37:34 发布

_天明_

最新推荐文章于 2020-08-30 11:37:34 发布

阅读量193

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/qq_32256033/article/details/106588796

版权

算法专栏收录该内容

35 篇文章 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==NULL) return NULL;
        if(pNode->right!=NULL){
            TreeLinkNode* t=pNode->right;
            while(true){
               
                if(t->left==NULL) return t;
                 t=t->left;
            }
            return t;
        }
        else{
            while(true){
                if(pNode->next==NULL) return NULL;
             if(pNode->next->left==pNode) return pNode->next;
                  pNode=pNode->next;
               
            }
            return NULL;
           
        }
    }
};