HDU4546 比赛难度

设置优先级队列

{

sum:当前和

nex:加入下个元素的和

ith:将要考虑的下个元素

}
以nex为优先级，小的先出队

读入数据后排序，初始化队列第一个元素(0,a[0],0)

每次出队一个元素，入队(sum,sum+a[ith],ith+1)，(nex,nex+a[ith],ith+1)，即是否加上a[ith]都考虑进去了。

这样每次新加入的元素都是下一个最小的(nex)，进行m次就得到了第m小。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<queue>
using namespace std;
int n, m;
const int maxn = 10011;
struct Node
{
    int sum;
    int nex;
    int ith;
    Node(){}
    Node(int sum_, int nex_, int ith_){sum = sum_, nex = nex_, ith = ith_;}
    bool operator<(const Node &b)const
    {return b.nex < nex;}
};
int a[maxn];
int Cal(int m)
{
    priority_queue<Node> q;
    Node lin;
    lin.sum = 0, lin.nex = a[0], lin.ith = 0;
    q.push(lin);
    int cnt = 0;
    a[n] = 0;
    while(cnt < m)
    {
        lin = q.top(); q.pop();
        if(lin.ith >= n) continue;
        q.push(Node(lin.sum, lin.sum + a[lin.ith + 1], lin.ith + 1));
        q.push(Node(lin.nex, lin.nex + a[lin.ith + 1], lin.ith + 1));
        cnt += 1;
    }
    for(m = 0; !q.empty(); m ++) a[m] = q.top().sum, q.pop();
    sort(a, a + m);
    return a[m - 1];
}

int main()
{
    int i, t, ca;
    for(scanf("%d", &t), ca = 1; ca <= t; ca ++)
    {
        scanf("%d%d", &n, &m);
        for(i = 0; i < n; i ++)
            scanf("%d", &a[i]);
        sort(a, a + n);
        printf("Case #%d: %d\n", ca, Cal(m));
    }
    return 0;
}