HDU4546 比赛难度 (优先队列)

题目链接:

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


题意:

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


分析:

对a[]序列进行排序

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

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

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


代码如下:


#include 
#include 
#include 
#include 
#include 
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 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


你可能感兴趣的:(数据结构)