0
点赞
收藏
分享

微信扫一扫

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

纽二 2022-05-01 阅读 40

剑指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;
    }
};
举报

相关推荐

0 条评论