首页 > 其他 > 详细

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

时间:2014-04-17 03:20:36      阅读:483      评论:0      收藏:0      [点我收藏+]

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!