剑指offer之面试题30最小的k个数

问题描述：

输入n个整数，找出其中最小的k个数。

例如：

输入4,5,1,6,2,7,3,8这8个数字，则找出最小的4个数字是1,2,3,4

实现代码如下：

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <limits.h>

int doPartition(int *a,int left,int right){
	if(a==NULL){
		return INT_MIN;
	}
	int part= left-1;
	int key = a[right];
	int i;
	for(i=left;i<right;i++){
		if(key>a[i]){
			part++;
			int temp = a[i];
			a[i]=a[part];
			a[part]=temp;
		}
	}
	a[right]=a[++part];
	a[part]=key;
	return part;
}

bool findMinK(int *a,int left,int right,int k){
	if(a==NULL){
		return false;
	}
	if(left==right){
		return true;
	}
	int part = doPartition(a,left,right);
	int i = part -left +1;
	if(k==i){
		return true;
	}else if(k>i){
		return findMinK(a,part+1,right,k-i);
	}else{
		return findMinK(a,left,part-1,k);
	}
}
int main(int argc, char *argv[])
{
	int a[]={
		1,3,2,4,6,5,7,9,8
	};
	int n =sizeof(a)/sizeof(int);
	int minK = 5;
	if(findMinK(a,0,n-1,minK)){
		int i;
		for(i=0;i<minK;i++){
			printf("%d\t",a[i]);
		}
	}
	return 0;
}

上面算法的期望时间复杂度是O(n)。

参考资料：
剑指offer

备注：
转载请注明出处：http://blog.csdn.net/wsyw126/article/details/51383656
作者：WSYW126