uva 11997 K smallest sums (优先队列 多路归并)

算法入门经典 训练指南 p189

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
struct Item
{
    int s,b;
    Item(int s,int b) :s(s),b(b) {}
    bool operator < (const Item&rhs) const {return s>rhs.s;}
};

void merge(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));
    }
}

const int maxn=768;
int A[maxn][maxn];

int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        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],n);

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