码迷,mamicode.com
首页 > 其他好文 > 详细

272. Closest Binary Search Tree Value II

时间:2016-10-13 07:44:58      阅读:187      评论: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
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!