牛客网 剑指offer 二叉树中序遍历的下一个结点

时间限制:1秒空间限制:32768K热度指数:4658

**算法知识视频讲解

题目描述

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

题解:有右孩子,则找右孩子的最左孩子,若无则返回右孩子

无右孩子,则返回父节点的左孩子。

代码如下:

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

打赏一个呗

取消

感谢您的支持,我会继续努力的!

扫码支持
扫码支持
扫码打赏,你说多少就多少

打开支付宝扫一扫,即可进行扫码打赏哦