编程之美 2.5 寻找最大的K个数(1)

/* top_K.c */
/* 编程之美 2.5 寻找最大的K个数 */

/* 二分搜索策略 */

//求数组中大于等于Vmid的个数
int f(int arr[], int N, int Vmid)
{
	int i;
	int count = 0;
	
	for (i = 0; i < N; i++)
	{
		if (arr[i] >= Vmid)
		{
			count++;
		}
	}
	
	return count;
}

// 寻找第K大的数
int find_max_K(int arr[], int N, int Vmax, int Vmin, int K)
{
	int Vmid;
	
	while (Vmax - Vmin > 1)
	{
		Vmid = Vmin + (Vmax - Vmin) * 0.5;
		if (f(arr, N, Vmid) >= K)
		{
			Vmin = Vmid;
		}
		else
		{
			Vmax = Vmid;
		}
	}
	
	return Vmin;
}

int main()
{
	int array[10] = {1, 5, 11, 34, 3, 9, 45, 7, 23, 10};
	int i;
	int Vmax, Vmin;
	int K = 4;
	int max_K;

	Vmax = Vmin = array[0];
	for (i = 0; i < 10; i++)
	{
		if (array[i] < Vmin)
		{
			Vmin = array[i];
		}
		if (array[i] > Vmax)
		{
			Vmax = array[i];
		}
	}

	max_K = find_max_K(array, 10, Vmax, Vmin, K);
	printf("%d\n", max_K);
	printf("result:");
	for (i = 0; i < 10; i++)
	{
		if (array[i] >= max_K)
		{
			printf("%d ", array[i]);
		}
	}

	printf("\n\n");

	return 0;
}