1167. Minimum Cost to Connect Sticks

Description

You have some number of sticks with positive integer lengths. These lengths are given as an array sticks, where sticks[i] is the length of the ith stick.

You can connect any two sticks of lengths x and y into one stick by paying a cost of x + y. You must connect all the sticks until there is only one stick remaining.

Return the minimum cost of connecting all the given sticks into one stick in this way.

Constraints

• 1 <= sticks.length <= 104

• 1 <= sticks[i] <= 104

Approach

Links

• Binarysearch

• GeeksforGeeks

• Leetcode

• ProgramCreek

• YouTube

Examples

Example 1

Input: sticks = [2,4,3]

Output: 14

Explanation: You start with sticks = [2,4,3].

1. Combine sticks 2 and 3 for a cost of 2 + 3 = 5. Now you have sticks = [5,4].

2. Combine sticks 5 and 4 for a cost of 5 + 4 = 9. Now you have sticks = [9].

There is only one stick left, so you are done. The total cost is 5 + 9 = 14.

Example 2

Example 3

Solutions

Solution 1

/**
 * Time complexity : O(NlogN), where N is the length of the input array. 
 *    Let's break it down:
 *    Step 1) Adding N elements to the priority queue will be O(NlogN).
 *    Step 2) We remove two of the smallest elements and then add one element 
 *        to the priority queue until we are left with one element. Since each 
 *        such operation will reduce one element from the priority queue, 
 *        we will perform N−1 such operations. Now, we know that both add 
 *        and remove operations take O(logN) in priority queue, therefore, 
 *        complexity of this step will be O(NlogN).
 * Space complexity : O(N) since we will store N elements in our priority queue.
 */

class Solution {
    public int connectSticks(int[] sticks) {
        if(sticks == null || sticks.length < 2) {
            return 0;
        }
        Queue<Integer> pq = new PriorityQueue<>();
        for(int stick: sticks) {
            pq.offer(stick);
        }
        
        int totalSticks = 0;
        while(pq.size() > 1) {
            int sum = pq.poll() + pq.poll();
            totalSticks += sum;
            pq.offer(sum);
        }
        
        return totalSticks;
    }
}

Follow up