POJ-2442 Sequence （K路归并问题拓展）

题意：有n行含m个元素序列，从每行取一个数得到他们的和，一共可以得到m^n个和。输出前n个最小的和。

思路：可以用优先队列递归解决，当只取前两行的数，得到两个数的和的前n小的序列。这个序列就相当于把第一行和第二行合并，再解决n-1行的子问题。

用优先队列解决的时候也有点小技巧，类似尺取法。

//236 KB	563 ms	C++	1480 B	
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;


struct node
{
    int val;
    int a,b;
    bool operator <(const node &a) const
    {
        return val>a.val;
    }
};

priority_queue<node>que;
const int N = 2000+20;
int a[N],b[N];
int tmp[N];
int n,m;

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++) scanf("%d",&a[i]);
        sort(a+1,a+1+m);
        for(int i=1;i<n;i++)
        {

            for(int i=1;i<=m;i++)
                scanf("%d",&b[i]);
            sort(b+1,b+1+m);
            while(!que.empty()) que.pop();
            for(int i=1;i<=m;i++)
            {
                node t;
                t.val=a[i]+b[1];
                t.a=i;
                t.b=1;
                que.push(t);
            }

            for(int i=1;i<=m;i++)
            {

                node cur=que.top();
                que.pop();
                tmp[i] = cur.val;
                node nxt;
                nxt.val=a[cur.a]+b[cur.b+1];
                nxt.a = cur.a;
                nxt.b = cur.b+1;
                que.push(nxt);

            }
            for(int i=1;i<=m;i++)
                a[i]=tmp[i];
        }
        for(int i=1;i<=m;i++)
            printf("%d%c",a[i],i==m? '\n':' ');
    }
    return 0;
}