NC16663 合并果子

题目

• 原题地址：合并果子
• 题目编号：NC16663
• 题目类型：队列、堆
• 时间限制：C/C++ 1秒，其他语言2秒
• 空间限制：C/C++ 131072K，其他语言262144K

1.题目大意

• n堆果子，需要两两合并直至只剩一堆，每次消耗的体力为两堆果子的质量之和，求消耗的最小体力

2.题目分析

• 优先队列试一下，每次合并质量最小的两堆果子，就过了

3.题目代码

#include <bits/stdc++.h>

using namespace std;

struct cmp{
    bool operator()(pair<int,int> a, pair<int,int> b) {
        return a.first > b.first;
    }
};

int main() {
    int n;
    cin >> n;
    priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> q;
    for(int i=1;i<=n;i++) {
        int tmp;
        cin >> tmp;
        q.push({tmp, i});
    }
    int ans = 0;
    while(q.size()!=1){
        pair<int, int> p1 = q.top();
        q.pop();
        pair<int, int> p2 = q.top();
        q.pop();
        ans += p1.first+p2.first;
        q.push({p1.first+p2.first, 0});
    }
    cout << ans << endl;
}