剑指offer08二叉树的下一个节点.

最新推荐文章于 2024-10-14 17:32:24 发布

Engineering Lion

最新推荐文章于 2024-10-14 17:32:24 发布

阅读量532

点赞数

分类专栏： leetcode刷题文章标签：二叉树后继节点中序遍历算法数据结构

本文链接：https://blog.csdn.net/qq_23844501/article/details/124524989

版权

leetcode刷题专栏收录该内容

8 篇文章 0 订阅

订阅专栏

剑指offer08二叉树的下一个节点.

/*方法：看当前节点是否有右孩子
 *  有：下一个节点就是右孩子的最左节点
 *  无：当前节点是否有父节点
 *      是：是不是父节点的右孩子：
 *           是：指针变换，往上追溯
 *           否：父节点就是下一个节点
 *       否：回复NULL
 */
//代码进行了整合，但是思路就是这个
class Solution {
public:
    TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
        if(!root) return root;
        TreeNode* ret;
        TreeNode* CurrentNode = p;
        TreeNode* pparentNode = p->pparent;

        //p节点存在右孩子,求右孩子的最左节点
        if(CurrentNode->right){
            CurrentNode = CurrentNode -> right;
            while(CurrentNode -> left){
                CurrentNode = CurrentNode -> left;
            }
            ret = CurrentNode;
        }else{
            while((pparentNode) && (CurrentNode == pparentNode->right)){
                CurrentNode = pparentNode;
                pparentNode = pparentNode ->pparent;
            }
            //不论是父节点等于了NUll，还是当前节点是父节点的左子树，都成立。
            ret = pparentNode;
        }
        return ret;
    }
};