LeetCode 1167. 连接棒材的最低费用（优先队列+贪心）

1. 题目

为了装修新房，你需要加工一些长度为正整数的棒材 sticks。

如果要将长度分别为 X 和 Y 的两根棒材连接在一起，你需要支付 X + Y 的费用。

由于施工需要，你必须将所有棒材连接成一根。

返回你把所有棒材 sticks 连成一根所需要的最低费用。注意你可以任意选择棒材连接的顺序。

示例 1：
输入：sticks = [2,4,3]
输出：14
解释：先将 2 和 3 连接成 5，花费 5；再将 5 和 4 连接成 9；总花费为 14。
示例 2：
输入：sticks = [1,8,3,5]
输出：30
 
提示：
1 <= sticks.length <= 10^4
1 <= sticks[i] <= 10^4

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-cost-to-connect-sticks
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• 短的优先，后面会被多次加总，短的优先保证花费最小

class Solution {
public:
    int connectSticks(vector<int>& sticks) {
        priority_queue<int,vector<int>,greater<int>> q;//小顶堆
        for(int i : sticks) 
        	q.push(i);
        int a, b, sum = 0;
        while(q.size() > 1)
        {
        	a = q.top();//取出2根短的
        	q.pop();
        	b = q.top();
        	q.pop();
        	q.push(a+b);//粘好了
        	sum += a+b;//花费
        }
        return sum;
    }
};

396 ms 23 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！