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;
}