算法：O(n)时间内在数组中找到第k个小的数

关注

VIP免费原创最新推荐文章于 2026-06-23 10:57:59 发布 2k 阅读 AI 写同款· GEO 优化 ›

无序数组中找到最小的k个数 O(N)解法快速排序原理方法和BFPRT方法最简便的方法是循环K次，依次找出最小的，第2小的，第K小的值，时间复杂度O(N*K)，这种方法使用于K比较小的时候，如个位数，如果N数据量很大，K也较大，不适用。因为K大于N时，时间复杂度变成O(N^2)。第二种是堆排序，建立一个大根堆，时间复杂度是O(NlogK)，此方法博客中有写到，详见堆排序。一、用快速排序，随机取枢轴量可以做到平均时间复杂度O(N)，但是值选的不好会退化到O(N^2... 阅读详情

从一个长度为n的无序数组中，找到第k小的那个数，被称为k-select问题。我们经常见到的问题：找到中位数，找到第二个大的数，都是这个问题的特例。

算法的复杂度是O(20n），当选择为5个数为一个小集合时。

OJ风格的完整代码如下：

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<assert.h>

#define Q 5
#define LOCAL

int A[1001];
int n;
int k;


int cmp(const void *a, const void *b) {
	return *(int *)a - *(int *)b;
}

int trivivalSelect(int *A, int n, int k) {
	int temp[1001];
	memset(temp, 0, sizeof(int)*1001);
	memmove(temp, A, sizeof(int)*n);
	qsort(temp, n, sizeof(int), cmp);
	printf("trivivalSelect: %d\n", temp[k]);
	return temp[k];
}


int kselect(int* A, int n, int k) {
	if (n < Q) {
		return trivivalSelect(A, n, k);
	}
	int mid[1001];
	int temp[1001];
	memset(mid, 0, sizeof(int)*1001);


	
	int i = 0;
	int c = 0;
	while (i < n) {
		int count = (n - i < Q ? n - i : Q);
		memset(temp, 0, sizeof(int)*1001);
		memmove(temp, A + i, sizeof(int)*count);
		qsort(temp, count, sizeof(int), cmp);
		mi