期望时间为线性时间的选择算法(C++)

#include <iostream>
using namespace std;

void exchange(int& x, int& y){
    int tem = x;
    x = y;
    y = tem;
}

 int par(int a[],int b,int e){
    int x = a[b];
    int j = b + 1;
    int i = b;
    for(;j != e + 1;++j){
        if(a[j] <= x){   
            ++i;
            exchange(a[i],a[j]);
        }
    }
    exchange(a[b],a[i]);
    return i;
}

int select(int a[],int p,int r,int i){// To select the i.th little digit

    if(p == r)
        return a[r];

    int q = par(a,p,r);//q是划分后首元素在序列中的下标

    int k = q - p + 1;//k是已经确定此序列a[p...r]的第k小元素 
    if(i == k)
        return a[q];
    else if(i < k)
        return select(a,p,q - 1,i);
    else
        return select(a,q + 1,r,i - k);//第一次把q写成k了。。。

}  

int main(int argc, char *argv[])
{
    int a[] = {50,2,11,22,33,44,66,88,77,99};

    cout << select(a,0,9,-1) << endl;
    return 0;
}

posted on 2015-09-01 19:31  泉山绿树  阅读(34)  评论(0)    收藏  举报

导航