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