1167. 连接棒材的最低费用

1167. 连接棒材的最低费用

为了装修新房，你需要加工一些长度为正整数的棒材 。棒材以数组 sticks 的形式给出，其中 sticks[i]是第 i 根棒材的长度。

如果要将长度分别为 x 和 y 的两根棒材连接在一起，你需要支付 x + y 的费用。 由于施工需要，你必须将所有棒材连接成一根。

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

示例 1：

输入：sticks = [2,4,3]
输出：14
解释：从 sticks = [2,4,3] 开始。
1. 连接 2 和 3 ，费用为 2 + 3 = 5 。现在 sticks = [5,4]
2. 连接 5 和 4 ，费用为 5 + 4 = 9 。现在 sticks = [9]
所有棒材已经连成一根，总费用 5 + 9 = 14

解法：小顶堆

思路

高端的名词，哈夫曼编码。好的你现在懂了。

其实就是个贪心，先把两个小元素合成一个大元素，再把这个大元素放在一起，再从里面挑俩个小元素，直到合成一个大元素。

用一个优先队列（堆）就搞定了。

复杂度

时间复杂度：$O(NlogN)$，建堆需要 $O(NlogN)$，push 和 pop 分别要 $O(log)$，需要 push 和 pop 共 $3(n-1)$ 次。

空间复杂度：$O(1)$

代码

class Solution {
    public int connectSticks(int[] s) {
        PriorityQueue<Integer> q = new PriorityQueue<> ();
        for (int x : s) q.add(x);
        int res = 0;
        while (q.size() > 1) {
            int x = q.poll();
            int y = q.poll();
            res += x + y;
            q.add(x + y);
        }
        return res;
    }
}