【题】NKOJ 3765 k个最小和

NKOJ 3765 k个最小和
时间限制 : - MS 空间限制 : 65536 KB 该题不设总的时间（空间）限制，每组测试数据具有独立的时空限制
评测说明 : 时限1000ms

问题描述
有k个整数数组，各包含k个元素，从每个数组中选取一个元素加起来，可以得到k^k个和，求这些和中最小的k个值。

输入格式
第一行，一个整数k(k<=500)
接下来k行，每行k个正整数(<=1000000)

输出格式
一行，k个有小到大排列的整数，表示最小的k个和

样例输入 1
3
1 8 5
9 2 5
10 7 6

样例输出 1
9 10 12

样例输入 2
2
1 1
1 2

样例输出 2
2 2

提示
样例1说明：
选出的三组数分别是(1 2 6) (1 2 7) (1 5 6)

来源 改编自uva11997

思路：
先考虑两个数组两两合并得到小k个数的情况，将第三个数组与前面合并得的数组再次合并，便可以解决问题。
两两合并的方法：
a、b排序后，先将a[i]+b[1]的值压入堆中。每次取出一个最小值，作为加入新序列中，再将a[i]+b[j+1]的值压入堆中。

小优化：
只存b的编号，a[i]+b[j+1]的值为 当前值-b[j]+b[j+1];

代码：

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
const int need=502;

struct fy
{
    int num,id;
    fy(int a,int b) {num=a,id=b;}
    friend bool operator< (fy a,fy b)
    {
        return a.num>b.num;
    }
};

int k;
int a[need]={0,1,2,3,4,5,6,7,8},b[need]={0,1,2,3,4,5,6,7,8};

void merge()
{
    priority_queue<fy> q;
    if(k==0) puts("dddd");
    for(int i=1;i<=k;i++) 
    q.push(fy(a[i]+b[1],1));
    for(int i=1,num,id;i<=k;i++)
    {
        num=q.top().num,id=q.top().id;
        q.pop();
        a[i]=num;
        if(id<k) q.push(fy(num-b[id]+b[id+1],id+1));
    }
}

int main()
{
    scanf("%d",&k);
    for(int i=1;i<=k;i++) scanf("%d",&a[i]);
    sort(&a[1],&a[k+1]);
    for(int i=1,j;i<k;i++)
    {
        for(j=1;j<=k;j++) scanf("%d",&b[j]);
        sort(&b[1],&b[k+1]);
        merge();
    }
    for(int i=1;i<=k;i++) printf("%d ",a[i]);
}