码迷,mamicode.com
首页 > 其他好文 > 详细

POJ-2442 Sequence (K路归并问题拓展)

时间:2015-03-30 13:28:22      阅读:151      评论:0      收藏:0      [点我收藏+]

标签:

题意:有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;
}


POJ-2442 Sequence (K路归并问题拓展)

标签:

原文地址:http://blog.csdn.net/kalilili/article/details/44747961

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!