二叉树的后继节点

最新推荐文章于 2024-07-21 22:06:29 发布

luck++

最新推荐文章于 2024-07-21 22:06:29 发布

阅读量512

点赞数 1

分类专栏：算法文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/qq_61344207/article/details/130833281

版权

算法专栏收录该内容

14 篇文章 0 订阅

订阅专栏

Successor node

问题

求二叉树里一个节点的后继节点

思路

判断情况：

节点有右子树：找右子树的最左节点

节点没有右子树：往上查找，但节点不能为父节点的左子树

实现

class Node
{
    public:
    Node* left;
    Node* right;
    Node* parent;
    Node():left(nullptr),right(nullptr),parent(nullptr){}   
};

Node* GetSuccessorNode(Node* root,Node* cur)
{
    if(!cur) return cur;
    
    if(cur->right) return GetLeftestNode(cur->right);
    Node* parent = cur->parent;
    while(parent != nullptr && parent->left != cur)
    {
        cur = parent;
        parent = cur->parent;
    }
    return parent;
}

Node* GetLeftestNode(Node* root)
{
    while(root->left)
    {
        root = root->left;
    }
    return root;
}