最小的K个数

题目
输入n个整数,找出其中最小的k个数。例如,输入4、5、1、6、2、7、3、8这8个数,则最小的4个数字是1,2,3,4.
方法一:快排
如果基于数组的第k个数字来调整,则使得比第k个数字小的所有数字都位于数组的左边,比第k个数字大的所有数字都位于数组的右边。这样调整之后,位于数组中左边的k个数字就是最小的k个数字(这k个数字不一定是排序的)。
源代码

package Arithmetic;

import java.util.ArrayList;

public class GetLeastKNumbers {
    public static ArrayList<Integer> GetLeastNumbers(int[] nums, int k) {
        ArrayList<Integer> ret = new ArrayList<>();//同来保存最小的k个数
        if (k > nums.length || k <= 0)
            return ret;
        findKthSmallest(nums, k);//排序后修改了数组中数字的前后顺序
        for (int i = 0; i < k; i++)
            ret.add(nums[i]);
        return ret;
    }

    private static void findKthSmallest(int[] nums, int k) {
        int i = 0, j = nums.length - 1;
        while (i < j) {
            int l = partition(nums, i, j);//找基准数的位置
            if (l == k) {//找到k直接跳出
                break;
            }
            if (l < k) {//小于
                i = l + 1;
            } else {//大于
                j = l - 1;
            }
        }
    }

    //快排
    private static int partition(int[] nums, int i, int j) {
        int tmp = nums[i];//基准数
        while (i < j) {
            while (i < j && nums[j] > tmp) j--;//队尾元素大于基准数
            nums[i] = nums[j];//如果队尾元素小于基准数,则将其赋值给i
            while (i < j && nums[i] < tmp) i++;//队首元素小于基准数
            nums[j] = nums[i];//如果队首元素大于基准数,则将其赋值给j
        }
        nums[i] = tmp;//基准数的位置
        return i;//返回基准数的索引
    }

    public static void main(String[] args) {
        System.out.println(GetLeastNumbers(new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9}, 7));
    }
}

方法二:堆排
创建一个大小为k的数据容器来存储最小的k个数字,接下来每次从输入的n个整数中读入一个数。如果容器中已有的数字少于k个,则直接把这次读入的整数放入容器中;如果容器中已有k个数字了,也就是容器已满,此时我们不能再插入新的数字而只能替换已有的数字。找出这已有的k个数中的最大值,然后拿这次待插入的整数和最大值进行比较。如果待插入的值比当前已有的最小值小,则用这个数替换当前已有的最大值;如果待插入的值比当前已有的最大值还大,那么这个数不可能是最小的k个整数之一,于是我们可以抛弃这个整数。
源代码

import java.util.PriorityQueue;

public class GetLeastKNumbers {
	public static ArrayList<Integer> GetLeastNumbers1(int[] nums, int k) {
        if (k > nums.length || k <= 0)
            return new ArrayList<>();
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>((o1, o2) -> o2 - o1);//堆排序,只能保证根是最大(最小),整个堆不是有序的
        for (int num : nums) {
            maxHeap.add(num);
            if (maxHeap.size() > k) {
                maxHeap.poll();
            }
        }
        return new ArrayList<>(maxHeap);
    }
     public static void main(String[] args) {
        System.out.println(GetLeastNumbers1(new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9}, 7));
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值