标签：index right get list mid II lintcode901 树中 left

二叉搜索树中最接近的值 II

题目：二叉搜索树中最接近的值 II

给定一棵非空二叉搜索树以及一个target值，找到 BST 中最接近给定值的 k 个数。

示例：

输入:
{3,1,4,#,2}
0.275000
2
输出:
[1,2]
解释：
二叉树 {3,1,4,#,2}，表示如下的树结构：
  3
 /  \
1    4
 \
  2

题解：二分法+双指针

1. 分治法将二叉树所有节点存储在集合中

2.二分法找到target在集合中应该插入的下标

3.从插入的下标开始，利用双指针依次找到最接近的k个值。

public class Solution {
    List<Integer> list;

    public void preOrder(TreeNode root) {
        if (root == null) return;
        preOrder(root.left);
        list.add(root.val);
        preOrder(root.right);
    }

    public List<Integer> closestKValues(TreeNode root, double target, int k) {
        list = new ArrayList<>();
        List<Integer> res = new ArrayList<>();
        preOrder(root);
        int length = list.size();

        //求得插入点
        int mid = 0, left = 0, right = length - 1;
        while (left < right) {
            mid = (left + right) / 2;
            if (target < list.get(mid)) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }

        left = mid;
        right = mid + 1;
        int index = 0;
        
        //双指针求最接近的k个值
        while (left >= 0 && right < length && index < k) {
            double leftDValue = Math.abs(target - list.get(left));
            double rightDValue = Math.abs(target - list.get(right));
            if (leftDValue < rightDValue) {
                res.add(list.get(left));
                left--;
                index++;
            } else {
                res.add(list.get(right));
                right++;
                index++;
            }
        }

        if (index < k) {
            if (left >= 0) {
                while (left >= 0 && index < k) {
                    res.add(list.get(left));
                    left--;
                    index++;
                }
            } else {
                while (right < length && index < k) {
                    res.add(list.get(right));
                    right++;
                    index++;
                }
            }
        }
        return res;
    }
}

标签：index,right,get,list,mid,II,lintcode901,树中,left
来源： https://www.cnblogs.com/greengages/p/15758053.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。