PTA修理牧场c++版——山东科技大学

在这里插入图片描述输入样例:

8
4 5 1 2 1 3 1 1

输出样例:

49

虽说这道题贴上了树的标签,但我好像没有用到树的知识hh
stl大法好

#include<bits/stdc++.h>
using namespace std;
int n;
int a[10001];
priority_queue <int,vector<int>,greater<int> > q;
int sum=0;
void init()
{
    
    
    cin>>n;
    for(int i=0;i<n;i++)
    {
    
    
        cin>>a[i];
        q.push(a[i]);
    }

}
void solve()
{
    
    
    while(!q.empty())
    {
    
    
        if(q.size()==1)
        {
    
    
            cout<<sum<<endl;
            break;
        }
        int x1=q.top();
        q.pop();
        int x2=q.top();
        q.pop();
        sum+=(x1+x2);
        //cout<<sum<<endl;
        q.push(x1+x2);
    }
}
int main()
{
    
    
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    init();
    solve();
}

创作不易
每天进步一点点,十天进步十点点,加油!
更多PTA代码请到我的博客里参考

ps:代码仅供参考,请勿抄袭

猜你喜欢

转载自blog.csdn.net/scorpion_legend/article/details/109132690