刚看题的时候想用贪心,结果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;
}