(六)线性选择

来源:互联网 发布:梦想小镇mac同步手机 编辑:程序博客网 时间:2024/06/17 09:32
/************************************************以期望线性时间做选择:返回数组A[p,r]中第i小的数算法通过分治递归划分,像快排那样,将数组划分成两边,然后根据情况,处理划分后的某一边,算法伪代码如下:select(A,p,r,i)    if p=r        then return A[p]    q<——partion(A,p,r)    k<——q-p+1    if i=k //第i小的数恰好在为分割点        then return A[q]    else if i<k   //第i小的数在左边部分        then return select(A,p,q-1,i)    else         // 第i小的数在右边部分        return select(A,q+1,r,i-k)************************************************/#include <stdio.h>#define MAX_INT ~(1<<31)int partition(int *arr,int low,int high){    int pivot=arr[high];    int i=low-1;    int j,tmp;    for(j=low;j<high;++j)        if(arr[j]<pivot){            tmp=arr[++i];            arr[i]=arr[j];            arr[j]=tmp;        }    tmp=arr[i+1];    arr[i+1]=arr[high];    arr[high]=tmp;    return i+1;}int select(int *arr,int p,int r,int i){    if(arr==NULL || p<0 || p>r)        return MAX_INT;    if(p==r)        return arr[p];    int q=partition(arr,p,r);    int k=q-p+1;    if(i==k)        return arr[q];    else if(i<k)        return select(arr,p,q-1,i);    else         return select(arr,q+1,r,i-k);}int main(){    int arr[10]={1,4,6,2,5,8,7,6,9,12};    printf("the 5th number is:%d\n",select(arr,0,9,5));}

原创粉丝点击