算法学习(一)归并排序求解N个数中最大的K个数

描述

给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。

数据范围:0\le k,n \le 100000≤k,n≤10000,数组中每个数的大小0 \le val \le 10000≤val≤1000

要求:空间复杂度 O(n)O(n) ,时间复杂度 O(nlogk)O(nlogk)

我的解法:

        采用归并排序的思想,将数组分解成若干数组,再排序,取出前K个数

        也可使用快排,但不稳定,归并排序比较稳定,采用空间换时间的办法

//分治将input排序,取前K个
    ArrayList<Integer> temp = new ArrayList<>();
    public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
        if(input.length<=0||k<=0)return temp;
        a(input, 0, input.length - 1,new int[input.length]);
        for(int i=0;i<k;i++){
            temp.add(input[i]);
        }
        return temp;
    }
    public void a(int[] nums, int left, int right, int[] temp) {
        if (left == right) { //当拆分到数组当中只要一个值的时候,结束递归
            return;
        }
        int mid = (left + right) / 2; //找到下次要拆分的中间值
        a(nums, left, mid, temp); //记录树左边的
        a(nums, mid + 1, right, temp); //记录树右边的

        //合并两个区间
        for (int i = left; i <= right; i++) {
            temp[i] = nums[i];
        }
        int i = left;       //给辅助数组里面的值标点
        int j = mid + 1;
        for (int k = left; k <= right ; k++) {//k 就为当前要插入的位置
            if (i == mid + 1) {//此时左边的已经走完了,继续覆盖右边
                nums[k] = temp[j];
                j++;
            } else if (j == right + 1) {//此时右边的已经走完了,继续覆盖左边
                nums[k] = temp[i];
                i++;
            } else if (temp[i] <= temp[j]) {
                nums[k] = temp[i];
                i++;
            } else {
                nums[k] = temp[j];
                j++;
            }
        }
    }

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值