【算法实验一】--【递归分治法】--求第k小数

1574.求第k小数

时限：1000ms 内存限制：10000K  总时限：3000ms

描述

求第k小数

 

输入

先输入一个小于10000的正整数n，再输入n个整数，最后输入一个小于等于n的正整数k，

 

输出

输出其中第k小的数。

 

输入样例

5
2 98 34512 8492 1000
2

 

输出样例

98

解析：这是个线性时间选择题，基本思想是分治和递归。第一次将数组划分成两个数组a[first:mid]和a[mid+1:last]，使得前者的每个元素都不大于后者的，然后分别对两者进行相同的操作，划分直到具体的一个数，就使得数组被按从小到大排好了序。基本思想其实跟快速排序没有区别，只是操作略有不同，而且复杂度是O(n)，比快排的O(nlogn)更高效。代码如下:

#include <iostream>
using namespace std;

int a[10001];
void swap(int &a,int &b)
{
    int temp=a;
    a=b;
    b=temp;
}
int partition(int a[],int first,int last)
{
    int x=a[last];
    int middle=first;
    int j;
    for(j=first;j<last;j++)
    {//j从本次传过来的数的第一个到最后一个，如果哪个数小于最后一个数且不是第一个，就把它与第一个数字交换
        if(a[j]<x)
        {
            if(j!=middle)
              swap(a[middle],a[j]);
            middle++;//交换了之后middle++，结果就成为所有小于最后一个数的所有数都在前面middle个
        }
    }
    swap(a[middle],a[j]);//结束之后，这个区间里从第一个开始前middle个都小于最后一个数，然后交换，最后得到这个区间的序列前middle个是小的，后面的每个数都小于后面的数字
    return middle;
}
void select(int a[],int first,int last)
{
    if(first<last)//一直分，直到把他们排序完毕
    {
        int mid=partition(a,first,last);//每次的界线由partitionh函数决定
        select(a,first,mid-1);
        select(a,mid+1,last);
    }
}
int main()
{
    int n,k;
    cin>>n;
    for(int i=0;i<n;i++)
        cin>>a[i];
    cin>>k;
    select(a,0,n-1);
    cout<<a[k-1]<<endl;
    return 0;

}