寻找最小K个数算法-优快云博客

本文介绍了一种通过快速选择算法查找整数数组中最小K个数的方法。该算法使用了分区技术，并提供了完整的Java实现代码。对于输入的整数数组，此方法能够有效地找出并返回最小的K个数。

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

题目描述

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

import java.util.ArrayList;
import java.util.Scanner;

public class Solution {
   public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
		int index;
		ArrayList<Integer> list = new ArrayList<Integer>();
		int start = 0;
		int end = input.length - 1;
		
		if(k>input.length||input.length == 0 || k<= 0){
			return list;
		}
       	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++) {
			list.add(input[i]);
		}
		return  list;
	}

	private int partition(int[] input, int low, int hight) {
		int point = input[low];
		while (low < hight) {
			while (low < hight && input[hight] >= point) {
				hight--;
			}
			input[low] = input[hight];
			while (low < hight && input[low] <= point) {
				low++;
			}
			input[hight] = input[low];
		}
		input[low] = point;
		return low;
	}
}