避坑详解最小的k个数！！（剑指offer java版）

最新推荐文章于 2022-07-12 19:26:10 发布

orchid96

最新推荐文章于 2022-07-12 19:26:10 发布

阅读量150

点赞数

分类专栏：剑指offer 文章标签：算法快速排序数据结构 java

本文链接：https://blog.youkuaiyun.com/doupudou6162/article/details/105908917

版权

剑指offer 专栏收录该内容

43 篇文章

订阅专栏

本文介绍了一种使用快速排序算法寻找数组中最小的K个数的方法，并提供了详细的算法实现代码。通过快速排序，可以有效地对数组进行排序，然后直接获取前K个元素作为最小的K个数。

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

最小的k个数

题目描述

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

思路

使用快速排序将数组进行排序，然后输出前k个数即可。
避坑：注意要加判断，对输入的k可能会出现越界的情况，比如k=0和k>input.length的情况要单独列出！！！

import java.util.ArrayList;
import java.util.Arrays;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
       // Arrays.sort(input);
        QuickSort(input,0,input.length-1);
        ArrayList<Integer> list = new ArrayList<>();
        if(k == 0  || k > input.length){
            return list;
        }else{
            for(int i = 0;i<k;i++){
            list.add(input[i]);
           }
        }       
        return list;
    }
    public static void QuickSort(int[] arr,int low ,int high){
        int temp,i,j,t;
        if(low>high){
            return;
        }
        i = low;
        j = high;
        temp = arr[low];
        //先看右边，依次往左递减
        while(i<j){
            while(arr[j]>=temp && i<j ){
                j--;
            }
            //再看左边，依次往右递增
            while(arr[i]<=temp && i<j){
                i++;
            }
            //如果满足条件则交换
            if(i < j){
                t = arr[j];
                arr[j] = arr[i];
                arr[i] = t;
            }            
        }
         //最后将基准为与i和j相等位置的数字交换
        arr[low] = arr[i];
        arr[i] = temp;
        //递归调用左半数组
        QuickSort(arr,low,i-1);
        //递归调用右半数组
        QuickSort(arr,i+1,high);
    }
}