UVA 11997 K Smallest Sums 优先队列

刚看题的时候想用贪心，结果wa了，然后就没有思路了。。。

看了别人的做法后，用归并，结果超时。。（觉得应该是O(k*k*logk)啊... ）

然后看了别人的代码后才知道可以用优先队列优化。

从前m行里面选出k个最小的m个数的和，前m+1行选出的k个数一定是第m+1行的k个数和前m行选出的k个数两两之和的最小的k个（貌似说的不是很清楚。。）

然后就从第1行开始到第k-1行一直和第0行合并。

觉得如果用归并的话并入那里时间效率可能会快些。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<queue>
#include<map>
#include<set>
using namespace std;

#define mxn 1000
#define inf 0x3f3f3f3f
#define eps 1e-8
#define LL long long 
#define ULL unsigned long long
#define MP make_pair
struct node {
	int sum, last;
	node(){}
	node( int _sum, int _last ): sum( _sum ), last( _last ) {};
	bool operator < ( const node &b ) const {
		return sum > b.sum;
	}
};
int a[mxn][mxn];
int k;
void merge( int *x ) {
	priority_queue< node > q;
	for( int i = 0; i < k; ++i )
		q.push( node( a[0][i] + x[0], 0 ) );
	for( int i = 0; i < k; ++i ) {
		node t = q.top();
		q.pop();
		a[0][i] = t.sum;
		t.sum = t.sum - x[t.last] + x[t.last+1];
		++t.last;
		q.push( t );
	}
}
int main() {
//	freopen( "test.txt", "r", stdin );
	while( scanf( "%d", &k ) != EOF ) {
		for( int i = 0; i < k; ++ i )  {
			for( int j = 0; j < k; ++j )
				scanf( "%d", &a[i][j] );
			sort( a[i], a[i] + k );
		}
		for( int i = 1; i < k; ++i )
			merge( a[i] );
		for( int i = 0; i < k; ++i )
			printf( "%d%c", a[0][i], i == k - 1 ? '\n' : ' ' );
	}
	return 0;
}