更多精彩内容,请关注【力扣简单题】。
题目
难度:★★☆☆☆
类型:数组,益智
有一堆石头,每块石头的重量都是正整数。
每一回合,从中选出两块最重的石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下,就返回 0。
提示
1 <= stones.length <= 30
1 <= stones[i] <= 1000
解答
由于每次只需要考虑最大的两块石头,因此我们在一开始把输入的石头从小到大排序,拿出来其中最大的两块,然后计算抵消结果,如果不是零,把抵消完的小块插入到排序石头数组中该有的位置,使整体依旧有序,然后继续重复上述操作,直到石头数组中的石头个数不大于1为止。
from bisect import bisect_left as locate
class Solution:
def lastStoneWeight(self, stones):
"""
:param stones: List[int]
:return: int
"""
stones.sort()
while len(stones) > 1:
stone_1, stone_2 = stones.pop(), stones.pop()
if stone_1 != stone_2:
stone_rest = stone_1 - stone_2
stones.insert(locate(stones, stone_rest), stone_rest)
return stones[0] if len(stones) > 0 else 0
如有疑问或建议,欢迎评论区留言~