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