class Solution {
public:
int lastStoneWeight(vector<int>& stones) {
priority_queue<int> PQ(stones.begin(), stones.end());
while(PQ.size() > 1) {
int y = PQ.top();
PQ.pop();
int x = PQ.top();
PQ.pop();
if(x != y) PQ.push(y - x);
}
if(PQ.empty()) return 0; // 这里需要注意可能全部被清空,返回 0
return PQ.top();
}
};
【leetcode 优先队列(堆) C++】1046. Last Stone Weight
最新推荐文章于 2024-10-05 12:07:50 发布