LeetCode 1167. 连接棒材的最低费用 (优先级队列 + 哈夫曼编码思想)

最新推荐文章于 2023-11-29 22:59:41 发布

原创最新推荐文章于 2023-11-29 22:59:41 发布 · 344 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 专栏收录该内容

40 篇文章

订阅专栏

本文介绍了一种算法问题，即如何以最低成本将多根不同长度的棒材通过连接组合成一根长棒。通过优先队列实现每次选取最短的两根棒材进行连接，以此递归直至所有棒材被连接成一根。

1167. 连接棒材的最低费用

为了装修新房，你需要加工一些长度为正整数的棒材 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


class Solution {
public:
    int connectSticks(vector<int>& sticks) {
    	int res = 0;
        priority_queue<int,vector<int>,greater<int> > Q;
        int n = sticks.size();
        for(int i = 0; i < n; i++){
        	Q.push(sticks[i]);
        }
        if(Q.size() == 1) return Q.top();
        while(Q.size() > 1){
        	int a = Q.top(); Q.pop();
        	int b = Q.top(); Q.pop();
        	Q.push(a+b);
        	res += (a+b);
        }
      	return res;
    }
};