【LeetCode】 1046. Last Stone Weight

Description

We have a collection of stones, each stone has a positive integer weight.

Each turn, we choose the two heaviest stones and smash them together. Suppose the stones have weights x and y with x <= y. The result of this smash is:

If x == y, both stones are totally destroyed;
If x != y, the stone of weight x is totally destroyed, and the stone of weight y has new weight y-x.
At the end, there is at most 1 stone left. Return the weight of this stone (or 0 if there are no stones left.)

Note:
1 <= stones.length <= 30
1 <= stones[i] <= 1000

我們有一堆石頭，每顆石頭都有一個正整數代表它的重量。

每一回合，我們選兩顆最大的石頭並讓它們相撞。假設那兩顆石頭的重量分別為x和y且x <= y，那麼相撞後的結果會是：

如果x == y，兩顆石頭都完全粉碎。
如果x != y，重量為x的石頭會完全粉碎，而重量為y的石頭重量會改為y - x。
最後只剩下一顆石頭時，回傳那顆石頭的重量（如果沒有剩下石頭就回傳0）。

提示：
1 <= stones.length <= 30
1 <= stones[i] <= 1000

Example:

Example 1:

Input: [2,7,4,1,8,1]
Output: 1
Explanation: 
We combine 7 and 8 to get 1 so the array converts to [2,4,1,1,1] then,
we combine 2 and 4 to get 2 so the array converts to [2,1,1,1] then,
we combine 2 and 1 to get 1 so the array converts to [1,1,1] then,
we combine 1 and 1 to get 0 so the array converts to [1] then that's the value of last stone.

Solution

• 作法很直覺，找到最大的兩顆，相撞，直到剩一顆。
• 找最大的兩顆可以跑一次迴圈就完成，也可以用排序來找。
• 使用排序可以確保最大的一定在最前面兩個，但因為相撞完之後都要再排序一次，因此超級慢。
• 比較快的方法是直接找最大值和次大值，C++裡面有max_element可以使用。

---

• 有看到該題目有提到可以用heap，感覺也是很聰明的方法，不過這邊就沒有做了。

Code

• 版本一

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        while(stones.size() > 1)
        {
            sort(stones.rbegin(), stones.rend());
            stones.push_back(abs(stones[0] - stones[1]));
            stones.erase(stones.begin(), stones.begin() + 2);
        }
        if(stones.empty())
            return 0;
        return stones[0];
    }
};

• 版本二

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        while(stones.size() > 1)
        {
            auto it1 = max_element(stones.begin(), stones.end());
            int r = *it1;
            stones.erase(it1);
            auto it2 = max_element(stones.begin(), stones.end());
            r -= *it2;
            stones.erase(it2);
            stones.push_back(r);
        }
        if(stones.empty())
            return 0;
        return stones[0];
    }
};

tags: LeetCode C++