【精】LintCode领扣算法问题答案：448. 二叉查找树的中序后继

最新推荐文章于 2022-08-09 19:09:05 发布

二当家的白帽子

最新推荐文章于 2022-08-09 19:09:05 发布

阅读量720

点赞数 1

分类专栏： LintCode领扣算法题目

本文链接：https://blog.csdn.net/leyi520/article/details/114536059

版权

二叉查找树中序遍历后继节点算法数据结构

关键词由CSDN通过智能技术生成

LintCode领扣算法题目专栏收录该内容

279 篇文章 17 订阅

订阅专栏

448. 二叉查找树的中序后继

描述

给定一个二叉查找树(什么是二叉查找树)，以及一个节点，求该节点在中序遍历的后继，如果没有则返回null

保证p是给定二叉树中的一个节点。(您可以直接通过内存地址找到p)

样例 1：

输入: 
	{1,#,2}, node with value 1
输出: 
	2
解释: 
	  1
	   \
	    2

样例 2：

输入: 
	{2,1,3}, node with value 1
输出: 
	2
解释: 
	    2
	   / \
	  1   3

挑战

O(h)，其中h是BST的高度。

原题传送门

题解

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */


public class Solution {
    /*
     * @param root: The root of the BST.
     * @param p: You need find the successor node of p.
     * @return: Successor of p.
     */
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        // write your code here

        TreeNode pre = null;
        TreeNode cur = root;
        while (cur != null) {
            if (cur == p) {
                if (cur.right != null) {
                    TreeNode ans = cur.right;
                    while (ans.left != null) {
                        ans = ans.left;
                    }
                    return ans;
                }
                return pre;
            }
            if (p.val > cur.val) {
                cur = cur.right;
            } else {
                pre = cur;
                cur = cur.left;
            }
        }

        return null;
    }
}