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