算法-排序k路平衡归并与败者树

Posted 2020-11-21 yangyuliufeng

const int MaxValue=999;   //根据实际情况选择最大值
void kwaymerge(Element *r,int k){
    int i,q;
    r=new Element[k];   //在败者树中的k个记录
    int *key=new int[k+1];  //k个排序码和建树单元key[k]
    int *loser=new int[k];  //k-1个败者和冠军loser[0]
    for(i=0; i<k; i++){      //从k个归并段输入第一个记录及其排序码
        InputRecord(r[i]);
        key[i]=r[i].key;
    }
    for(i=0; i<k; i++) loser[i]=k;   //败者树所有结点都赋值k，表示第k归并段
    key[k]=-MaxValue;
    for (i=k-1; i>=0; i--) adjust(key,loser,k,i);   //从key[k-1]起~key[0]起调整形成败者树
    while(key[loser[0]]!=MaxValue){     //当MaxValue上升到loser[0]归并完毕
        q=loser[0];    //取当前最小关键码所在归并段段号送q
        OutputRecord(r[q]);   //r[q]写到输出归并段
        InputRecord(r[q]);    //从第q个归并段再读入下一条记录
        key[q]=r[q].key;
        adjust(key,loser,k,q);  //从key[q]起调整为败者树
    }
    Output end of run marker;   //输出段结束标志
    delete []r;
    delete []k;
    delete []loser;
}

void adjust(int key[]; int loser[]; int k; int q){
    for(int t=(k+q)/2; t>0; t/=2)
        if(key[loser[t]]<key[q]){
            int temp=q;
            q=loser[t];
            loser[t]=temp;
        }
    loser[0]=q;
}