剑指offer -- 最小的k个数

最新推荐文章于 2024-04-23 21:26:48 发布

tkzc_csk

最新推荐文章于 2024-04-23 21:26:48 发布

阅读量150

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签：剑指offer

本文链接：https://blog.youkuaiyun.com/tkzc_csk/article/details/83417104

算法专栏收录该内容

212 篇文章

订阅专栏

本文介绍了一种寻找数组中最小K个数的有效算法，包括利用快速排序思想和最大堆的方法，这两种方法都能高效地解决该问题，适用于大数据量的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

AC代码1

利用快速排序的思想找到第k - 1位置的数字。
缺点：改动了原数组。

import java.util.ArrayList;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> resList = new ArrayList<>();
        //必须保证k <= input.length
        if(input == null || input.length == 0 || k <= 0 || k > input.length) return resList;
        int start = 0;
        int end = input.length - 1;
        int index = partition(input, start, end);
        while(index != k - 1){
            if(index > k - 1) {
                end = index - 1;
                index = partition(input, start, end);
            }
            else{
                start = index + 1;
                index = partition(input, start, end);
            }
        }
        for(int i = 0;i < k;i++) resList.add(input[i]);
        return resList;
    }
    
    private int partition(int[] array, int start, int end){
        int temp = array[start];
        int low = start;
        int high = end;
        while(low < high){
            while(high > low && array[high] >= temp) high--;
            if(low < high) array[low] = array[high];
            while(high > low && array[low] <= temp) low++;
            if(low < high) array[high] = array[low];
        }
        array[low]。 = temp;
        return low;
    }
}

方法2

利用最大堆。