leetcode658:找到K个最接近的元素

找到K个最接近的元素

给定一个 排序好 的数组 arr ,两个整数 kx ,从数组中找到最靠近 x(两数之差最小)的 k 个数。返回的结果必须要是按升序排好的。

整数 a 比整数 b 更接近 x 需要满足:

  • |a - x| < |b - x| 或者
  • |a - x| == |b - x|a < b
public List<Integer> findClosestElements(int[] arr, int k, int x) {
        if(arr.length == 1){
            return new ArrayList<>(Arrays.asList(arr[0]));
        }
        int end = arr.length-1;
        int start = 0;
        if(arr[start] >= x){
            return Arrays.stream(Arrays.copyOfRange(arr, 0, k)).boxed().collect(Collectors.toList());
        }
        if(arr[end] <= x){
            return Arrays.stream(Arrays.copyOfRange(arr, arr.length-k,arr.length)).boxed().collect(Collectors.toList());
        }
        while(true){
            boolean flag = true;
            if(arr[start+1] <= x){
                start++;
                flag = false;
            }
            if(arr[end-1] >= x){
                end--;
                flag = false;
            }
            if(flag){
                break;
            }
        }
        return getClose(start,end,x,k,arr);
    }

    private List<Integer> getClose(int start, int end, int x, int k, int[] arr) {
        ArrayList<Integer> res = new ArrayList<Integer>(k);
        while(end - start -1 != k){
            if(end == arr.length || (start != -1 && arr[end] - x >= x - arr[start])){
                start--;
            }else{
                end++;
            }
        }
        return Arrays.stream(Arrays.copyOfRange(arr, start + 1, end)).boxed().collect(Collectors.toList());
    }

写不明白,最优解以后再补

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值