Zznu 1913: yifan and matrix (多路归并)

题目链接：

　　1913: yifan and matrix

题目描述：

　　有一个n*n的矩阵，在每一行取出一个数，可以得到n个数的和，问前n小的和分别是多少？

解题思路：
　　对于两个数组a[n],b[n]，我们可以用二路归并维护一个升序序列a[i]+b[j](1<=i<=n,1<=j<=n)，先把a[i]+b[1](1<=i<=n)加进优先队列，每次取出队首元素，记录下来，然后再改变为a[i]+b[j+1]再加进队列.

依次进行n-1二路合并即可。

　　优美的代码将要闪亮登场！！！！！

#include <queue>
#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 500;
struct node
{
    int sum, num;
    node (int a, int b):sum(a),num(b){}
    bool operator < (const node &a) const {
        return sum > a.sum;
    }
};
int maps[maxn][maxn], ans[maxn], arr[maxn];
int n;
void Merge (int cnt)
{
    priority_queue <node> Q;
    for (int i=0; i<n; i++)
        Q.push (node(ans[i]+maps[cnt][0], 0));
    for (int i=0; i<n; i++)
    {
        node p = Q.top ();
        Q.pop();
        arr[i] = p.sum;
        if (p.num != n)
        {
            p.sum += maps[cnt][p.num+1] - maps[cnt][p.num];
            p.num ++;
            Q.push (p);
        }
    }
    for (int i=0; i<n; i++)
        ans[i] = arr[i];
}
int main ()
{
    while (scanf ("%d", &n) != EOF)
    {
        for (int i=0; i<n; i++)
        {
            for (int j=0; j<n; j++)
                scanf ("%d", &maps[i][j]);
            sort (maps[i], maps[i]+n);
        }
        for (int i=0; i<n; i++)
            ans[i] = maps[0][i];
        for (int i=1; i<n; i++)
            Merge (i);
        for (int i=0; i<n; i++)
            printf ("%d%c", ans[i], i==n-1?'
':' ');
    }
    return 0;
}