最小的K个数

最新推荐文章于 2025-04-07 10:55:57 发布

原创最新推荐文章于 2025-04-07 10:55:57 发布 · 197 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用快速排序思想来寻找数组中最小的K个整数的方法。该算法通过部分排序，避免了完全排序的高时间复杂度，提高了效率。

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

题目描述

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

快速排序

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> al=new ArrayList<Integer>();
        if(input.length < k || k <= 0||input.length ==0)
            return al;
        int low=0;
        int high=input.length-1;
        int index=partition(input,low,high);
        while(index!=k-1){
            if(index>k-1){
                high=index-1;
                index=partition(input,low,high);
            }
            else{
                low=index+1;
                index=partition(input,low,high);
            } 
        }
        for(int i=0;i<k;i++){
            al.add(input[i]);
        }
        return al;
    }
    public static int partition(int[] data,int low,int high){
        int temp=data[low];
        while(low<high){
            while(temp<=data[high]&&low<high){
                high--;
            }
            data[low]=data[high];
            while(temp>=data[low]&&low<high){
                low++;
            }
            data[high]=data[low];
        }
        data[low]=temp;
        return low;
    }
}