uva 11997 STL 优先队列priority_queue +简单数学

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 +简单数学)