k路求和归并

K路求和归并：
两路归并过程：
1.将两个序列升序排列。
2.将一个序列B的首与另外一个序列A的每一个元素求和，加入到一个优先队列中。
3.依次取优先队列队头，查找队头来自B的哪一个位置，将现在和中的B[n]替换为B[n-1]
时间复杂度O(N*logN)

K路归并则进行K-1次两路归并
如果数组长度不同，将长的放到前面可以节约时间（启发式合并）

---

UVA 11997
题意：求K个数组求和合并的前K个小的和

#include <algorithm>
#include <iostream>
#include <queue>
#include <cstdio>
#include <vector>
using namespace std;
const int maxn = 800;

struct state{
    int s, b;
    bool operator < (const state &st) const{
        return s > st.s; /// initial bigger priority
    }
};

int n;
int A[maxn][maxn];

void Merge(int A[], int B[], int res[])
{
    priority_queue<state> q;
    for(int i = 0; i < n; i++){
        state temp;
        temp.s = A[i] + B[0];
        temp.b = 0;
        q.push(temp);
    }

    for(int i = 0; i < n; i++){
        state cur = q.top(); q.pop();
        res[i] = cur.s;
        int b = cur.b;
        if(b + 1 < n){
            cur.s = cur.s - B[b] + B[b+1];
            cur.b = b+1;
            q.push(cur);
        }
    }
}

int main()
{
    while(~scanf("%d", &n)){
        for(int i = 0; i < n; i++){
            for(int j = 0; j < n; j++){
                scanf("%d", &A[i][j]);
            }
            sort(A[i], A[i]+n);
        }
        for(int i = 1; i < n; i++)
            Merge(A[0], A[i], A[0]);

        printf("%d", A[0][0]);
        for(int i = 1; i < n; i++){
            printf(" %d", A[0][i]);
        }printf("\n");
    }
    return 0;
}

转载于:https://www.cnblogs.com/shh00/p/7265871.html