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

文章目录

    • 1. 题目
    • 2. 解题

1. 题目

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

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

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

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

示例 1:
输入:sticks = [2,4,3]
输出:14
解释:先将 23 连接成 5,花费 5;再将 54 连接成 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阿明),一起加油、一起学习进步!
Michael阿明

你可能感兴趣的:(LeetCode)