C++实现在二叉树中找到一个节点的后继节点

在二叉树中找到一个节点的后继节点

1. 题目描述

给定一颗二叉树和其中的一个节点，如果找出中序遍历的下一个节点？树中的节点除了有两个分别指向左、右子节点的指针，还有一个指向父节点的指针。

已知该二叉树如图所示：

2. 思路

分为三种情况：

1. 一个节点有右孩子，则在中序遍历中，该节点的后继是它的右子树的最左节点。

2. 这个节点是它父亲的左孩子，则该节点的后继节点是它的父亲。

3. 这个节点是它父亲的右孩子，则需要一直向上搜索，直到它们n-1代祖先是它第n代祖先的左孩子，则它的后继就是第n个祖先。如果一直搜索到根节点，也没有找到n-1代祖先是它第n代祖先的左孩子，则该节点是整个树的中序遍历中的最后一个节点，即它没有后继。

首先该二叉树的数据结构：

struct Node {
    int val;
    Node* left;
    Node* right;
    Node* parent;
    Node (int val):
        val(val), left(nullptr),
        right(nullptr), parent(nullptr) {}
};

完整代码：

#include <iostream>

struct Node {
    int val;
    Node* left;
    Node* right;
    Node* parent;
    Node (int val):
        val(val), left(nullptr),
        right(nullptr), parent(nullptr) {}
};

Node* getMostLeftNode(Node* node) {
    if (node == nullptr) {
        return node;
    }
    while (node->left != nullptr) {
        node = node->left;
    }
    return node;
}

Node* getNextNode(Node* node) {
    if (node == nullptr) {
        return node;
    }
    if (node->right != nullptr) {   // if node has right tree, we find the leftest node
        return getMostLeftNode(node->right);
    } else {
        Node* parent = node->parent;
        while (parent != nullptr && parent->left != node) {
            node = parent;
            parent = node->parent;
        }
        return parent;
    }
}

int main()
{
    Node* head = new Node(6);
    head->parent = nullptr;
    head->left = new Node(3);
    head->left->parent = head;
    head->left->left = new Node(1);
    head->left->left->parent = head->left;
    head->left->left->right = new Node(2);
    head->left->left->right->parent = head->left->left;
    head->left->right = new Node(4);
    head->left->right->parent = head->left;
    head->left->right->right = new Node(5);
    head->left->right->right->parent = head->left->right;
    head->right = new Node(9);
    head->right->parent = head;
    head->right->left = new Node(8);
    head->right->left->parent = head->right;
    head->right->left->left = new Node(7);
    head->right->left->left->parent = head->right->left;
    head->right->right = new Node(10);
    head->right->right->parent = head->right;

    Node* test1 = head->left->left;
    std::cout << test1->val <<  " next node: " <<  getNextNode(test1)->val << std::endl;
    Node* test2 = head->left->left->right;
    std::cout << test2->val <<  " next node: " <<  getNextNode(test2)->val << std::endl;
    Node* test3 = head->right->right;
    std::cout << test3->val <<  " next node: " <<  getNextNode(test3)<< std::endl;
    return 0;
}

这里我们选择三个作为测试：

1. 有右子树比如1
2. 没有右子树比如2
3. 没有右子树比如这里最后一个节点10

3. 参考

1. 求一棵二叉树中任意节点的后继节点（后继节点是指在中序遍历中紧随其后的节点）

二叉树的前驱节点也是类似的。