首页 > 其他 > 详细

272. Closest Binary Search Tree Value II

时间:2016-10-13 07:44:40      阅读:231      评论:0      收藏:0      [点我收藏+]

Given a non-empty binary search tree and a target value, find k values in the BST that are closest to the target.

Note:

  • Given target value is a floating point.
  • You may assume k is always valid, that is: k ≤ total nodes.
  • You are guaranteed to have only one unique set of k values in the BST that are closest to the target.

 

Follow up:
Assume that the BST is balanced, could you solve it in less than O(n) runtime (where n = total nodes)?

Hint:

  1. Consider implement these two helper functions:
    1. getPredecessor(N), which returns the next smaller node to N.
    2. getSuccessor(N), which returns the next larger node to N.
    Show More Hint 

 

Hide Company Tags
 Google
Hide Tags
 Tree Stack
Show Similar Problems
 用inorder来做 , bst inorder是有序的。
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
 //use inorder it will be a increase array
public class Solution {
    public List<Integer> closestKValues(TreeNode root, double target, int k) {
        List<Integer> res = new LinkedList<Integer>();
        getClosestKValues(res, root, target, k);
        return res;
    }
    public void getClosestKValues(List<Integer> res, TreeNode root, double target, int k){
        if(root == null) return;
        getClosestKValues(res, root.left, target, k);
        if(res.size() == k){
            if(Math.abs(res.get(0) - target) > Math.abs(root.val - target)){
                res.remove(0);
                res.add(root.val);
            }
            else
                return;
        }
        else
            res.add(root.val);
        getClosestKValues(res, root.right, target, k);
    }
}

 

272. Closest Binary Search Tree Value II

原文:http://www.cnblogs.com/joannacode/p/5955051.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!