面试题58:二叉树的下一个节点

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

using namespace std;
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 Left(pNode->right);
        else if(pNode->next == NULL)    //根节点,且不含右子树,就没有下一个节点
            return NULL;
        else if(pNode->next->left == pNode)   //左子节点
            return pNode->next;
        else 
        {
            TreeLinkNode *temp = pNode;
            while(temp->next != NULL && temp->next->right == temp)
                temp=temp->next;
            if(temp->next == NULL)
                return NULL;
            else
                return temp->next;
        }
    }
    TreeLinkNode *Left(TreeLinkNode *node)
    {
        if(node->left == NULL)
            return node;
        return Left(node->left);
    }
};

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页