【王道数据结构】【chapter8排序】【P360t2】

试编写一个算法，使之能够在数组L[1……n]中找出第k小的元素（即从小到大排序后处于第k个位置的元素）（可以直接采用排序，但下面的排序的代码只是为了方便核对是不是第k小的元素，k从0开始计算）

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

int * testArray(int size)
{
    int a[100]={0};
    int *tmp=(int*) malloc(sizeof(int )*size);
    for(int i=0;i<size;i++)
    {
        int t=rand()%100;
        if(a[t]!=1) tmp[i]=t,a[t]=1;
        else i--;
    }
    return tmp;
}

void print(int *tmp,int size)
{
    printf("the array is:");
    for(int i=0;i<size;i++) printf("%3d",tmp[i]);
    puts("");
}

int _sort(int* array,int left,int right)
{
    int tmp=array[left];
    while(left<right)
    {
        while(left<right&&array[right]>=tmp) --right;
        array[left]=array[right];
        while(left<right&&array[left]<=tmp) ++left;
        array[right]=array[left];
    }
    array[left]=tmp;
    return left;
}
void sort(int * array,int left,int right)
{
    if(left<right) {
        int split = _sort(array, left, right);
        sort(array, left, split - 1);
        sort(array, split + 1, right);
    }
}

//第k小，就是确定第k个位置上的元素
int kth_elem(int * array,int left,int right,int k)
{
    int tmp=array[left];
    int l=left,r=right;
    while(l<r)
    {
        while(l<r&&array[r]>tmp) r--;
        array[l]=array[r];
        while(l<r&&array[l]<tmp) l++;
        array[r]=array[l];
    }
    array[l]=tmp;
    if(l==k) return array[l];
    else if(l>k) return kth_elem(array,left,l-1,k);//在左半边继续找第k小的元素
    else return kth_elem(array,l+1,right,k);
}
int main() {
    srand(time(nullptr));
    int *r1= testArray(10);
    int *r2=(int *) malloc(sizeof (int)*10);
    for(int i=0;i<10;i++)r2[i]=r1[i];
    print(r1,10);
    sort(r2,0,9);
    print(r2,10);
    printf("%3d\n", kth_elem(r1,0,9,4));
    return 0;
}