第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;}