数组中第k个数(《编程珠玑(续)》第3章)

本文链接：https://blog.csdn.net/u010585135/article/details/40869985

选择数组当中第k个数，时间复杂度O(n)，这个原先在《编程珠玑》中有过讨论，现在给出迭代版本。

注意测试用例，这里用了1-9的所有的排列，每个排列中都计算k=1-9，这样就能把所有情况都考虑进去，不用担心有某种特殊情况，这个算法不能用。

#include<iostream>
#include<algorithm>
using namespace std;

int select(int A[],int lo,int hi,int k)
{
	int s=lo;
	if(lo<=hi)
	{
		int temp=lo+rand()%(hi-lo+1);
		swap(A[lo],A[temp]);
		for(int i=lo+1;i<=hi;i++)//注意这里i<=h
		{
			if(A[i]<A[lo])
				swap(A[++s],A[i]);
		}
		swap(A[s],A[lo]);
		if(s+1>k)
			return select(A,lo,s-1,k);
		else if(s+1==k)
			return A[k-1];//注意是第k个数，下标为k-1
		else
			return select(A,s+1,hi,k);
	}
}

int select1(int A[],int lo,int hi,int k)
{
	while(lo<=hi)
	{
		int m=lo;
		swap(A[lo],A[lo+rand()%(hi-lo+1)]);
		for(int i=lo+1;i<=hi;i++)
		{
			if(A[i]<A[lo])
				swap(A[++m],A[i]);
		}
		swap(A[m],A[lo]);
		if(m+1==k)
			return A[k-1];
		else if(m+1>k)
			hi=m-1;
		else
			lo=m+1;
	}
}

int select2(int A[],int lo,int hi,int k)
{
	while(lo<=hi)
	{
		int m=lo;
		swap(A[lo],A[lo+rand()%(hi-lo+1)]);
		for(int i=lo+1;i<=hi;i++)
		{
			if(A[i]<A[lo])
				swap(A[++m],A[i]);
		}
		swap(A[m],A[lo]);
		if(m+1>=k)
			hi=m-1;//当m+1==k时，hi=m-1，此后hi不再改变
		else
			lo=m+1;
	}
	return A[hi+1];
}

int main()
{
	int A[]={1,2,3,4,5,6,7,8,9};
	int B[9];
	do
	{
		for(int i=0;i<9;i++)
			B[i]=A[i];
		for(int i=1;i<=9;i++)
		{
			if(select2(B,0,8,i)!=i)
				cout<<B[i-1]<<endl;
		}
	}while(next_permutation(A,A+9));

	system("pause");
	return 0;
}