面试题04.06.后继者(找出二叉树中指定节点的下一个节点)(java)

最新推荐文章于 2022-06-29 09:35:55 发布

wmtmw

最新推荐文章于 2022-06-29 09:35:55 发布

阅读量263

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/yinianxx/article/details/106520642

leetcode 专栏收录该内容

87 篇文章

订阅专栏

本文介绍了一种在二叉树中寻找指定节点中序后继的算法。通过判断目标节点是否存在右子树，以及与根节点的大小关系，实现高效查找。适用于计算机科学与数据结构学习。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目
在这里插入图片描述
具体代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        if(root == null){
            return root;
        }
        //如果p存在右子树，那后继节点是右子树的最左孩子
        if(p.right != null){
            TreeNode node = p.right;
            while(node.left != null){
                node = node.left;
            }
            return node;
        }
        //如果p不存在右子树
        //如果p小于root，node = root 那遍历node的左孩子，直到p等于node，那node的父节点就是p的后继节点
        //如果p大于root，node = root 遍历node的右孩子，直到p<node，为上一种情况
        else{
            TreeNode res = null;
            TreeNode node = root;
            while(node != p){
                if(p.val < node.val){
                    res = node;
                    node = node.left;
                }else{
                    node = node.right;
                }
            }
            return res;
        }
    }
}