二叉树的下一个结点

最新推荐文章于 2023-11-06 00:22:49 发布

black_hidden

最新推荐文章于 2023-11-06 00:22:49 发布

阅读量152

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/black_hidden/article/details/78327084

版权

leetcode 专栏收录该内容

84 篇文章 1 订阅

订阅专栏

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

思路：
1、当前节点有右子树，则下一个节点为右子树中最左节点，即右子树沿着左子节点指针一直走完的最后一个节点。
2、当前节点没有右子树，且为父节点的左子节点，父节点为其下一个节点。
3、当前节点没有右子树，且为父节点的右子节点，则沿着指向父节点的指针，向上找到第一个为父节点左子节点的节点，即为下一个节点。

/*
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 == nullptr) return nullptr;
        TreeLinkNode *next = nullptr;
        if (pNode->right != nullptr) {
            TreeLinkNode *pRight = pNode->right;
            while (pRight->left != nullptr) {
                pRight = pRight->left;
            }
            next = pRight;
        }
        else if (pNode->next != nullptr) {
            TreeLinkNode *parent = pNode->next;
            TreeLinkNode *pCurrent = pNode;
            while (parent != nullptr && pCurrent == parent->right) {
                pCurrent = parent;
                parent = parent->next;
            }
            next = parent;
        }
        return next;
    }
};