本文主要是介绍Leetcode 1046. 最后一块石头的重量,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目
有一堆石头,每块石头的重量都是正整数。
每一回合,从中选出两块 最重的 石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下,就返回 0。
示例:
输入:[2,7,4,1,8,1]
输出:1
解释:
先选出 7 和 8,得到 1,所以数组转换为 [2,4,1,1,1],
再选出 2 和 4,得到 2,所以数组转换为 [2,1,1,1],
接着是 2 和 1,得到 1,所以数组转换为 [1,1,1],
最后选出 1 和 1,得到 0,最终数组转换为 [1],这就是最后剩下那块石头的重量。
提示:
1 <= stones.length <= 30
1 <= stones[i] <= 1000
解题思路
本题和1049. 最后一块石头的重量 II虽然是一个系列,但不存在隶属关系,如stones =[4,3,4,3,2],按本题每一回合,从中选出两块 最重的 石头,得到的最终结果是2,但按1049.最后一块石头的重量 II 任意选取石头,同时返回此石头最小的可能重量,得到的最终结果为0.
本题可利用std::priority_queue的特性,每次取重量最大的两个,把差值push到priority_queue,直到priority_queue只剩下一个元素。std::priority_queue底层实现是大小堆,因此使用大小堆也能解决此题。
代码实现
class Solution {
public:int lastStoneWeight(vector<int>& stones) {priority_queue<int> pq(stones.begin(), stones.end());while (pq.size() > 1) {int x = pq.top();pq.pop();int y = pq.top();pq.pop();if (x != y) {pq.push(x - y);}}return pq.empty() ? 0 : pq.top();}
};
这篇关于Leetcode 1046. 最后一块石头的重量的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!