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

最新推荐文章于 2025-02-25 20:47:36 发布

原创最新推荐文章于 2025-02-25 20:47:36 发布 · 334 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一个算法问题：如何找到给定二叉树中某个节点在中序遍历下的下一个节点。该算法考虑了节点是否有右子树、是否为父节点的左孩子等条件，并提供了一个具体的实现方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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);
    }
};