关于top N问题

思路1:堆处理

    //最小k个数字
    public int[] smallestK(int[] arr, int k) {
        if (arr.length==0)return arr;
        // 初始化成小顶堆
        for (int i = arr.length / 2 - 1; i >=0; i--) {
            adjust(arr, i, arr.length);
        }
        int n = k - 1;
        //因为不需要保持前n的有序,所以只需要到n
        for (int i = arr.length-1; i >n ; i--) {
            swap(arr,0,i);
            adjust(arr,0,i);
        }
        return arr;
    }

    //堆排序的堆调整
    private void adjust(int[] arr, int index, int length) {
        int tmp = arr[index];
        for (int i = index * 2 + 1; i < length; i = i * 2 + 1) {
            if (i + 1 < length && arr[i] < arr[i + 1]) {
                i++;
            }

            if (tmp < arr[i]) {
                arr[index] = arr[i];
                index = i;
            } else {
                break;
            }
        }
        arr[index] = tmp;
    }

    private void swap(int[] arr, int idx1, int idx2) {
        int tmp = arr[idx1];
        arr[idx1] = arr[idx2];
        arr[idx2] = tmp;
    }

思路2:快速排序法分治思想

    //最小k个数字
    int bound;
    public int[] smallestK(int[] arr, int k) {
        if (arr.length == 0) return arr;
        bound=k;
        qs(arr,0,arr.length-1);
        return arr;
    }

    private void qs(int[] arr, int left, int right) {
        if (left>right||left>=bound)return;
        int i = left, j = right;
        int base = arr[i];
        while (i < j) {
            while (i < j && arr[j] >= base) j--;
            if (i<j){
                arr[i]=arr[j];
                i++;
            }
            while (i < j && arr[i] <= base) i++;
            if (i<j){
                arr[j]=arr[i];
                j--;
            }
        }
        arr[i]=base;
        if (i==bound)return;
        qs(arr,left,i-1);
        qs(arr,i+1,right);
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值