第K小数-快速排序分割函数变体

第K小数问题陆陆续续碰到过几次，从大三算法分析期末考试，到2013数据结构考研最后一题，第K小数反复出现着，但可能因为自己极懒，这么多年的遇见都没有正儿八经地解决这个问题，其实关于这个问题比较好的解法，老师告诉我们就是利用快速排序里的分割函数，把分割的主元放出来，根据主元的下标，小的往左找，大的往右边找，俨然就是分治法的步骤，今天突然想到了这个问题，急切地想把之解决，于是，想了又想，写了又写，百度了又百度，在自己智慧和别人智慧的结合下，一个菜鸟多年的困扰在几行代码中解决了，代码写的很简单，就只是把partition这个函数改变了一番

第K小数简介：顾名思义，就是找出一堆数里面第K个小的数，当然多数情况下，这对数是无序的，比如3,2,1,4,5,6,9,7,8这堆数里，第1小的数为1，第2小的为2，依次类推

话不多说，上传代码

#include<iostream>
#include<stdlib.h>

using namespace std;

void swap(int a[],int b[])
{
	int temp=*a;
	*a=*b;
	*b=temp;
}


int partition_k(int k,int a[],int first,int end)
{
	int i=first;
	int j=end;
	while(i<j)
	{
		while(i<j&&a[i]<a[j]) j--;
		if(i<j)
		{
			swap(a[i],a[j]);
			i++;
		}
		while(i<j&&a[i]<a[j]) i++;
		if(i<j)
		{
			swap(a[i],a[j]);
			j--;
		}
	}
	//return i;改写部分 
	if(k==j-first+1) return a[j];
	else if(k<j-first+1) return partition_k(k,a,first,j-1);
	else return partition_k(k-(j-first+1),a,j+1,end);
}

int main()
{
	int a[9]={3,2,1,4,5,6,9,7,8};
	for(int i=0;i<9;i++)
	cout<<a[i]<<" ";
	cout<<endl;
	cout<<partition_k(7,a,0,9);
	cout<<endl;

}