Uva 11997 多路归并

题目链接：https://uva.onlinejudge.org/external/119/11997.pdf

题意：

k*k的矩阵，从每一行中选一个元素加起来，可以得到 k^k个和，求前 k 个最小值。

分析：

先把表头都放到优先队列中，每出一个，就从相应的表后面加一个。

#include <bits/stdc++.h>

using namespace std;

const int maxn = 755;

int A[maxn][maxn];

struct Item {
    int s,b;
    bool operator < (const Item& rhs) const {
        return s > rhs.s;
    }
};

void merga(int* A,int* B,int* C,int n) {
    priority_queue<Item> Q;
    for(int i=0;i<n;i++) {
        Q.push((Item){A[i]+B[0],0});
    }
    
    for(int i=0;i<n;i++) {
        Item item = Q.top();Q.pop();
        C[i] = item.s;
        int b = item.b;
        if(b+1<n) Q.push((Item){item.s-B[b]+B[b+1],b+1});
    }
}

int main(int argc, char** argv) {
    int n;
    while(scanf("%d",&n)!=EOF) {
        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++) {
            merga(A[0],A[i],A[0],n);
        }
        
        printf("%d",A[0][0]);
        for(int i=1;i<n;i++) {
            printf(" %d",A[0][i]);
        }
        puts("");
        
    }
    return 0;
}