LeetCode658-找到k个最接近的元素

最新推荐文章于 2023-09-25 21:24:42 发布

CountingStars619

最新推荐文章于 2023-09-25 21:24:42 发布

阅读量1.8k

点赞数

分类专栏：力扣文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_44736853/article/details/121064357

版权

力扣专栏收录该内容

69 篇文章 1 订阅

订阅专栏

该博客主要讨论了一种算法问题，即如何在数组中找到最接近给定目标值的子数组。通过使用二分查找技术，可以高效地确定子数组的边界。如果目标值小于等于数组的首元素，返回前k个元素；如果大于等于尾元素，返回最后k个元素。否则，利用二分查找找到目标值附近的下标，然后调整左右边界来找到最接近的k个元素。算法的时间复杂度为O(log(n)+k)，空间复杂度为O(k)。

摘要由CSDN通过智能技术生成

思路：如果x小于等于首元素，返回前k个元素即可；如果x大于等于尾元素，返回最后k个元素即可；如果x值在数组首元素于为尾元素的值域中间，首先利用二分查找搜索到下的下标或比x大一点的元素下标index，让low= index左边第k-1个元素的下标，high = index右边第k-1个元素的下标，要查找的k个元素一定位于[low, high]。如果左边的元素更接近x，则high–,如果右边的元素更接近x，则low++。

代码：

class Solution {
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
        List<Integer> ret = Arrays.stream(arr).boxed().collect(Collectors.toList());
        int n = ret.size();
        //如果x <=首元素，则返回前k个元素
        if(x <= ret.get(0)){
            return ret.subList(0, k);
            // //如果x >= 尾元素，则返回后k个元素
        }else if(x >= ret.get(n-1)){
            return ret.subList(n - k, n);
            //x介于arr[0]和arr【n-】
        }else {
            //如果x在数组中，返回下标，否则返回-(插入键)-1,插入键为大于key的第一个元素
            int index = Collections.binarySearch(ret, x);
            //此时index= -(大于x的第一个数) - 1
            if(index < 0){
                //返回大于x的第一个数
                index = -index - 1;
            }
            //index - k - 1
            int low = Math.max(0, index - k );

            int high = Math.min(n - 1, index + k - 1);

            while (high - low > k - 1){
                if(ret.get(high) - x < x - ret.get(low)  ){
                    low ++;
                }else {
                    high--;
                }
            }
            return ret.subList(low, high + 1);
        }
    }
}