Leetcode 270. Closest Binary Search Tree Value

最新推荐文章于 2022-05-01 08:34:11 发布

原创最新推荐文章于 2022-05-01 08:34:11 发布 · 227 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉搜索树 #binary search #中序遍历

Leetcode 专栏收录该内容

278 篇文章

订阅专栏

该博客讨论了两种方法来解决在二叉搜索树（BST）中找到最接近目标值的节点。方法1通过中序遍历将BST转换为有序列表，然后使用自定义比较器找到最接近的值。方法2利用BST的性质，采用二分查找法迭代地逼近目标值。两种方法分别具有不同的时间和空间复杂度。

在这里插入图片描述
方法1: 其实我们只需要把这个bst从小到大排序好，存进一个list，然后遍历list比较大小就可以得出结果。要实现从小到大排序，只需要inorder traversal就行。时间复杂n，空间复杂n

class Solution {
  public void inorder(TreeNode root, List<Integer> nums) {
    if (root == null) return;
    inorder(root.left, nums);
    nums.add(root.val);
    inorder(root.right, nums);
  }

  public int closestValue(TreeNode root, double target) {
    List<Integer> nums = new ArrayList();
    inorder(root, nums);
    return Collections.min(nums, new Comparator<Integer>() {
      @Override
      public int compare(Integer o1, Integer o2) {
        return Math.abs(o1 - target) < Math.abs(o2 - target) ? -1 : 1;
      }
    });
  }
}

方法2: binary search。其实也很容易想到，因为是bst啊，十有八九要考binary search。时间复杂logn，空间复杂1.

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int closestValue(TreeNode root, double target) {
        int ans = root.val;
        while(true){
            ans = Math.abs(root.val-target) > Math.abs(ans-target) ? ans : root.val;
            if(root.val == target){
                return root.val;
            }else if(root.val > target && root.left != null){
                root = root.left;
            }else if(root.val < target && root.right != null){
                root = root.right;
            }else{
                break;
            }
        }
        return ans;
    }
}