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

来源：互联网发布：怎么限制软件联网编辑：程序博客网时间：2024/06/06 01:27

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

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

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

//236 KB563 msC++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;}

0 0