POJ 2442 Sequence (堆+K路归并)

题目：http://poj.org/problem?id=2442

 

思路：优先队列（堆）+K路归并

//开始之前先说个例题：

例题一：

 

 

 

例题二:（例题一的应用）

 

有了前面2个例题，思路就基本清晰了，然后就是要明白，m个有序表的前n个最小和可由m-1个有序表的qian那个最小和与第m个有序表形成。以此类推，其实就是例二的扩展算法。

 

开始忘记给他们排序、WA了一次。。。。。。

 

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <iomanip>
#include <climits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <algorithm>
#include <string>
#include <cstring>

using namespace std;

const double EPS = 1e-11;

void Swap(int &a,int &b){   int t=a;a=b;b=t; }
int Max(int a,int b)    {   return a>b?a:b;  }
int Min(int a,int b)    {   return a<b?a:b;  }

int a[2005],b[2005];
int n,m;
struct node
{
    int a,b;
    int key;
    friend bool operator < (node n1 ,node n2)　　　　　　//priority_queue需要的cmp函数
    {
        return n2.key<n1.key;
    }
};

void fuck()　　　　　　　　　　　　　　　　　　　　//K路归并求两个序列和过程
{
    node t;
    priority_queue < node , vector< node > > Q;
    for (int i=0;i<n;i++)
    {
        t.a=i;　　　　　　　　　　　　　　　　　　//t.a表示这个最值来自哪个表：a[i]+b[0],a[i]+b[1],...,a[i]+b[n]
        t.b=0;　　　　　　　　　　　　　　　　　　//t.b表示当前表已经选到了a[i]+b[t.b](刚开始第一个是0)
        t.key=a[i]+b[0];
        Q.push(t);
    }

    int c[2005];
    int k=0;
    while(k!=n)
    {
        t=Q.top();
        Q.pop();
        c[k++]=t.key;　　　　　　　　　　　　　　//把最小值放入新序列中
        t.b++;　　　　　　　　　　　　　　　　　　//此表向后推一个
        t.key=a[t.a]+b[t.b];　　　　　　　　　　
        Q.push(t);　　　　　　　　　　　　　　　 //把当前表的新元素重新加入到堆中　
    }

    for (int i=0;i<n;i++)
        a[i]=c[i];
    return ;
}

int main()
{
    int t;
    scanf("%d",&t);

    while(t--)
    {
        cin>>m>>n;
        for (int i=0;i<n;i++)
            scanf("%d",&a[i]);
        sort(a,a+n);
        for (int i=0;i<m-1;i++)
        {
            for (int j=0;j<n;j++)
                scanf("%d",&b[j]);
            sort(b,b+n);
            fuck();
        }

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