HDU 4546 比赛难度

来源:互联网 发布:gta5淘宝刷钱会封号吗 编辑:程序博客网 时间:2024/05/28 05:13

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4546

用优先队列维护数据,只需计算m次即可。

那么如何保证每次得到最小值。

先从小到大排序

例如当前和是sum,参与加和的最后一个数的下标是i;

那么最小值势必是sum+a[i+1]和sum+a[i+1]-a[i]中的一个。

那么就可以将sum+a[i+1]和sum+a[i+1]-a[i]都放进优先队列维护,每次取出队首,拓展两个值,进队,第m次出队的势必是第m小的数

下面是AC代码:

<pre name="code" class="cpp">#include<iostream>#include<queue>#include<algorithm>using namespace std;const int maxn=10005;int n,m,a[maxn];struct node{    int s,p;    friend bool operator <(node a,node b)    {        return a.s>b.s;    }};int solve(){    priority_queue<node>q;    node t;    a[n]=1<<30;//下标为i拓展的数不可能成为最小值    t.s=a[0];t.p=0;    q.push(t);    int sum,pos;    for(int i=1;i<=m;i++)    {        //cout<<"yes"<<endl;        t=q.top();q.pop();        sum=t.s;        pos=t.p;        t.s=sum+a[pos+1];        t.p=pos+1;        q.push(t);        t.s=sum+a[pos+1]-a[pos];        q.push(t);    }    return sum;}int main(){    int t,Case=1;    cin>>t;    while(t--)    {        cin>>n>>m;        for(int i=0;i<n;i++) cin>>a[i];        sort(a,a+n);        cout<<"Case #"<<Case++<<": "<<solve()<<endl;    }    return 0;}



0 0
原创粉丝点击