【leetcode】1167 连接棒材的最低费用（贪心，哈夫曼编码）

最新推荐文章于 2024-04-19 18:18:22 发布

原创最新推荐文章于 2024-04-19 18:18:22 发布 · 1.9k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#贪心 #哈夫曼编码

LeetCode 专栏收录该内容

297 篇文章

订阅专栏

这篇博客介绍了如何利用贪心算法和哈夫曼编码来解决LeetCode上的1167题，即连接棒材以达到最低费用。通过排序和贪心策略，找到最优连接顺序，从而降低总成本。文章提供了题目描述、示例及解题思路。

题目链接：https://leetcode-cn.com/problems/minimum-cost-to-connect-sticks/

题目描述

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

思路

// 贪心+哈夫曼编码
// 时间复杂度O(nlongn) 空间复杂度O(n)
class Solution {
public:
    int connectSticks(vector<int>& sticks) {
        multiset<int, less<int>> m; // 最小堆
        for(int n:sticks)
            m.insert(n);
        int ret = 0;
        while(m.size()>1){
            int a = *m.begin();m.erase(m.begin());   // 取出当前最小的两个数
            int b = *m.begin();m.erase(m.begin());
            ret += a + b;
            m.insert(a+b);
        }
        return ret;
    }
};