首页 > 代码库 > HDU4546 比赛难度 (优先队列)

HDU4546 比赛难度 (优先队列)

题目链接:

http://acm.hdu.edu.cn/showproblem.php?pid=4546


题意:

给你n个数,问这n个数的组合的和,第m小是多少


分析:

对a[]序列进行排序

我们用一个结构体来存储 当前的和与它马上要加上的一个值将其放入队列中,

那么我们用队首的元素加上队首元素要加的值得下一个放入队列中,弹出队首元素,

进行m次这样的操作就可以得到第m小值.


代码如下:


#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;

const int maxn = 10010;

int a[maxn];

struct elg{
    int start,i;
    elg(int a,int b):start(a),i(b){}
    bool operator < (const struct elg &tmp)const{
        return start + a[i] > tmp.start + a[tmp.i];
    }
};

int main()
{
    int n,m,t,cas=1;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        sort(a,a+n);
        priority_queue<elg> Q;
        Q.push(elg(0,0));
        int ans  = 0;
        while(m--){
            if(Q.empty()) continue;
            elg tmp = Q.top();
            ans = tmp.start +a[tmp.i++];
            Q.pop();
            if(tmp.i<n){
                Q.push(tmp);
                tmp.start = ans;
                Q.push(tmp);
            }
        }
        printf("Case #%d: %d\n",cas++,ans);
    }
    return 0;
}


HDU4546 比赛难度 (优先队列)