练习场acm 题目55 懒省事的小明

李祖林

于 2016-07-31 09:23:16 发布

阅读量451

点赞数

CC 4.0 BY-SA版权

分类专栏： stl

本文链接：https://blog.youkuaiyun.com/u014360070/article/details/52076985

stl 专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种使用优先队列（堆）解决特定问题的方法。该问题要求计算将数组中所有元素合并成单一元素所需的最小代价。通过实现自定义比较器并利用优先队列的特性，可以有效地求解这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解法：优先队列！

 
#include<bits/stdc++.h>
#define ll long long
using namespace std;
struct cmp
{
    bool operator ()(const int a,const int b) const
    {
        return a>b;
    }
};
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        ll n,a;
        cin>>n;
        priority_queue<ll,vector<ll>,cmp>pq;
        for(int i=0;i<n;i++)
        {
            cin>>a;
            pq.push(a);
        }
        if(n==1)
        {
            cout<<n<<"\n";
            continue;
        }
        ll s=0;
        for(ll i=0;i<n;i++)
        {
            ll sum=0;
            ll x=pq.top();
            pq.pop();
            sum+=x;
            x=pq.top();
            pq.pop();
            sum+=x;
            s+=sum;
            if(pq.empty())
                break;
            pq.push(sum);
        }
        cout<<s<<"\n";
    }
    return 0;
}