http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&category=497&problem=3148
暴力k的k次方的算法当然是肯定 超时
稍动点脑子,先算出最小的(两排中最小的相加肯定是),维护一个优先队列,边pop,边push可能的最小的,然后最先出来的肯定是最小的k个
顺便练习下怎么用STL的优先队列
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
//优先队列析构如果花时间的话,那么你就在一个函数里建队列,
//每次调用结束后自动析构,只不过把值存到一个全局变量或者
//传递过来的数组里
//uva 11997 优先队列练习
//2014.4.16
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <functional>
using namespace std;
const int MAXN = 768;
int k;
int a[MAXN][MAXN];
/*注意这个Elem怎么写的,c++我还是太弱菜了*/
struct Elem{
int s,b;
Elem(int s,int b):s(s),b(b){} /*注意写法*/
bool operator < (const Elem &t)const{
return s>t.s;
}
};
void Add(int j)
{
priority_queue<Elem>q;
int c[MAXN];
for(int i=0;i<k;i++)
{
q.push(Elem(a[0][0]+a[j][i],0));
}
for(int i=0;i<k;i++)
{
Elem t = q.top();
int b=t.b;
q.pop();
//a[0][i]=t.s;
c[i]=t.s;
q.push(Elem(t.s-a[0][b]+a[0][b+1],b+1));
}
for(int i=0;i<k;i++)
a[0][i]=c[i];
}
int main()
{
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++)
{
Add(i);
}
for(int i=0;i<k;i++)
{
i!=k-1?printf("%d ",a[0][i]):printf("%d\n",a[0][i]);
}
}
return 0;
}
uva 11997 STL 优先队列priority_queue +简单数学,布布扣,bubuko.com
uva 11997 STL 优先队列priority_queue +简单数学
原文:http://blog.csdn.net/u011026968/article/details/23877097