UVa 11997 K Smallest Sums

题意分析：

有K个整数数组，各包含K个元素。在每个数组中取一个元素加起来，可以得到K^K个和。求这些和中最小的K个值（重复的值算多次）。其实这是一道多路合并的问题。可以先考虑两两合并。给出两个长度为n的有序表A和B，即数组。分别在A和B中任一个数相加，可以得到n^2个和。求这些和中最小的n个和。我们可以把这n^2和组织成如下n个有序表：表1： A1+B1<<A1+B2<<A1+B3<<.... 表2：A2+B1<<A2+B2<<A2+B3<<....经分析可知，这道题考察的是优先队列。

代码如下：

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
struct node{
int sum,index;
node(int sum,int index) :sum(sum) , index(index) {
}

bool operator < (const node& no) const {
return sum>no.sum;
}
};
int a[100][100];
priority_queue<node> que;
void merg(int* A,int* B,int* C,int n) {
int i;
for(i=0; i<n; i++) que.push(node(A[i]+B[0],0));
for(i=0; i<n; i++) {
node no=que.top();
que.pop();
C[i] = no.sum;
if(no.index+1<n) {
que.push(node(no.sum-B[no.index]+B[no.index+1],no.index+1));
}
}
}
int main() {
int n,i,j;
cin>>n;
for(i=0; i<n; i++)
for(j=0; j<n; j++)
cin>>a[i][j];
sort(a[0],a[0]+n);
for(i=1; i<n; i++){
merg(a[0],a[i],a[0],n);
}
cout<<a[0][0];
for(i=1; i<n; i++)
cout<<' '<<a[0][i];
cout<<endl;
return 0;
}