分治法-选择问题

问题描述
–给出含有n个元素表A(1: n)，确定其中的第k小元素。

设计思路
–利用PARTITION过程。
第一次划分后，划分元素v测定在A(j)的位置上，则有j-1个元素小于或等于A(j)，且有n-j个元素大于或等于A(j)。
①若k=j，则A(j)即是第k小元素；

②若k<j，则第k小元素将出现在A(1:j-1)中；

③若k>j，则第k小元素将出现在A(j+1:n)中。

找第k小元素
procedure SELECT(A, n, k) //在数组A(1:n)中找第k小元素，并将之放在A(k)中。//
integer n, k, m, r, j;
m←1; r ←n+1; A(n+1) ←+∞ //A(n+1)被定义，并置为一大值，用于限界//
loop //在进入循环时，1≤m≤k≤r≤n+1 //
j ←r //将剩余元素的最大下标加1后置给j //
call PARTITION(m，j) //返回j,它使得A(j)是第j小的值//
case
:k=j: return
:k<j: r←j //j是新的上界//
:else: m←j+1 //k>j, j+1是新的下界//
endcase
repeat
end SELECT

 

使用二次取中规则得选择算法的说明性描述

procedure SELECT2(A,k,n) //在集合A中找第k小元素，使用两次取中规则//
① 若n≤r，则采用插入法直接对A分类并返回第k小元素
② 把A分成大小为r的n/r个子集合，忽略多余的元素
③ 设M={m1,m2,…mn/r}是n/r子集合的中间值集合
④ v←SELECT2(M, n/r2 , n/r)
⑤ j←PARTITION(A，v) //v作为划分元素，划分后j等于划分元素所在位置的下标//
⑥ case
:k=j: return(v)
:k<j: 设S是A(1:j-1)中元素的集合
return(SELECT2(S,k,j-1))
:else: 设R是A(j+1:n)中元素的集合
return(SELECT2(R,k-j,n-j))
endcase
end SELECT2

经过测试的代码

#include <iostream>
using namespace std;

/*
r是一个全程变量，其取值为大于1的整数
分别设置了r=5以及r=7的两种情况
*/
int r = 5;
//int r = 7;
/*
函数功能：交换两个整数
*/
void Swap(int &x,int &y)
{
    int temp = x;
    x = y;
    y = temp;
}

/*
插入排序，输入数组a，数组的下标从m变换到n
输出升序的数组
*/
void insertionsort(int a[],int m,int n)
{
    int i,j,key;
    for(i=m;i<n;++i)
    {
        key = a[i];
        for(j=i-1;j>=m && a[j]>key;--j)
            a[j+1] = a[j];
        a[j+1] = key;
    }
}


/*
函数功能：传入数组a,数组开始下标begin,end作为结束下标，
划分后end等于划分元素所在位置的下标
*/
int partition(int a[],int begin,int end)
{
    end--;
    int i = begin +1;
    int temp = a[begin];
    while(true)
    {
        while(true)
        {
            if(i<end&& a[i] <= temp)
                ++i;
            else
                break;
        }
        while(true)
        {
            if(a[end]>temp&& end>=0)
                --end;
            else
                break;
        }
        if(i<end)
            swap(a[i],a[end]);
        else
            return end;
    }

}
/*
将数组分组n/r中每组的中间数，交换位置
放在数组的前部
*/
void interchange(int a[],int i,int j)
{
    swap(a[i],a[j]);
}

/*
返回一个整数v，使得v∈[m,p]
且A(v)是A(m:p)中第k小元素
*/
int select(int a[],int m,int p,int k)
{
    int n,i;
    int v=0;
    while(1)
    {
        if(p-m <= r)
        {
            insertionsort(a,m,p);
            return m+k-1;
        }
        n = p-m;
        for(i=0;i<n/r;++i)
        {
            insertionsort(a,m+i*r,m+(i+1)*r);
            interchange(a,m+i,m+(i)*r+r/2);
        }
        if((n/r)%2 ==1)
            v = select(a,m,m+n/r,(1+(n/r)/2));
        else
            v = select(a,m,m+n/r,(n/r)/2);
        interchange(a,m,v);
        v = p;
        v = partition(a,m,v);
        if((v-m+1) == k)
            return v;
        else if((v-m+1) > k)
            p = v;
        else
        {
            k = k-(v-m+1);
            m = v+1;
        }
    }
}
int main()
{
    /*
    测试数据
    a数组个数大于r;
    b数组个数小于r;
    c数组个数比较大；
    */
    int i;
    int a[9] = {65,70,75,80,85,60,55,50,45};
    int b[4] = {5,3,1,2};
    int c[34] = {57,72,37,19,60,
                47,63,8,12,17,6,
                70,53,13,62,5,7,
                30,20,50,1,10,14,
                2,38,15,25,27,4,
                24,75,65,9,42};
    /*
    分别打印出三个数组，然后输出测试的结果
    */
    for(i=0;i<9;++i)
        cout<<a[i]<<" ";
    cout<<endl;
    cout<<"a array:The first small number K(4) : "<<endl;
    cout<<a[select(a,0,9,4)]<<endl<<endl;
    for(i=0;i<4;++i)
        cout<<b[i]<<" ";
    cout<<endl;
    cout<<"b array:The first small number K(3) : "<<endl;
    cout<<b[select(b,0,4,3)]<<endl<<endl;
    for(i=0;i<34;++i)
        cout<<c[i]<<" ";
    cout<<endl;
    cout<<"c array:The first small number K(10) : "<<endl;
    cout<<c[select(c,0,34,10)]<<endl;
    return 0;
}