找出第N小数算法

2019独角兽企业重金招聘Python工程师标准>>>

 输入一个序列，例如N个，求第m小的数字

例如 1,3,9

那么第一小是 1，第二小是3

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
void swap(int& a,int &b)
{
	int c=a;
	a=b;
	b=c;
}
int partition(int* array,int b,int e)
{
	int x=array[e];
	int low=b-1;
	for(int i=0;i<e;++i)
	{
		if(array[i]<x)
		{
			low++;
			swap(array[i],array[low]);
		}
	}
	low++;
	swap(array[low],array[e]);
	return low;
}

int random_select(int* array,int b,int e,int i)
{
	if(b==e) return array[b];
	int q=partition(array,b,e);
	int k=q-b+1;
	if(i==k) return array[q];
	if(i<k) return random_select(array,b,q-1,i);
	else return random_select(array,q+1,e,i-k);	
}

int main()
{
	int array[10];
	srandom(time(0));
	for(int i=0;i<10;++i)
	{
		array[i]=random()%255;
		printf("%d ",array[i]);
	}
	putchar('\n');
	printf("第3小的数字是 : %d",random_select(array,0,9,3));
	putchar('\n');
	return 0;
}

其中partition函数就是quicksort的分割函数，函数会产生一个返回值 q  满足 bbegin<= q <= eend ,且 在array[begin...q-1] < array[q] <array[q+1....end]

 利用这个特点，找出第m小的数字。

转载于:https://my.oschina.net/000quanwei/blog/551750