(第五天)昨天种花,今天来搬砖。
题目:
有一堆石头,每块石头的重量都是正整数。 每一回合,从中选出两块 最重的 石头,然后将它们一起粉碎。假设石头的重量分别为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
题目分析:老样子先来分析题目,按照题意
1.我们需要再这堆石头列表种找出最大的两块,x,y,且x <= y
2.然后将其相减(粉碎),如果y=x,相减后等于零,消失,相减后大于0,则把剩余部分加入石头堆列表
3.循环上两部直到石头堆列表只剩一个(不管它是不是0)
思路分析:从石头列表搬出最大两块,我们可以使用for循环遍历石头列表的每个元素,然后比较它们的大小
max_1, max_2 = 0, 0
for i in range(len(stones)):
if stones[i] > max_1:
max_1 = stones[i]
stones.remove(max_1) # 找到最大的石头后删除最的的石头
#下一个循环就是找第二大的
for i in range(len(stones)):
if stones[i] > max_2:
max_2 = stones[i]
我们也可以用sort()方法直接排序
stones.sort()
接下来就是将第一第二大的石头不断相减,直到只剩一个(可能为零),可以用while循环实现
先写出大致框架
def lastStoneWeight(stones):
stones.sort() # 将列表从小到大排列
while True: # 循环
if len(stones) >= 2:
x = stones[-2] # 取出倒数第二个,也就是第二大的石头
y = stones[-1] # 取出倒数第一个,也就是第一大的石头
stones.pop(-1) # 删除最大的石头
stones.pop(-2) # 删除第二大的石头
stone = y - x # 最大的石头减第二大的石头
stones = append(stone) # 将粉碎后剩下的石头加入石头堆
return stone[0] # 返回最后剩下的石头
else: # stones列表长度为一的情况
return stones[0]
上面的代码还有一些问题,例如
1.当你删除最大的那块石头后,第二大的石头就成为列表中最大的石头
如stones=[3,2],搬出第一大还有stones=[2],这时候再有stones.pop(-2)删除倒数第二个元素就会出错,因为没有到第二个
2.return都是在while循环体内,因为没有借书条件程序将没完没了的进行。运行一会就能听见你电脑风扇散热的声音了
优化一下
def lastStoneWeight(stones):
stones.sort()
if len(stones) == 1: # 将原本的if,else,位置调换当列表长度为1时直接输出
print(stones[0])
else:
while True: # 将while循环放入else语句下
y = stones[-1]
stones.pop(-1) # 搬出最大的,第二大变成最大的
x = stones[-1]
stones.pop(-1) # 搬出最大的(第二大)
stone = y - x
stones.append(stone)
stones.sort() # 从新按小到大排序列表
return stones[0]
if len(stones) == 1: # 这样while循环就有了停止的条件
# 即列表长度 = 1时停止
break
return stones[0]
效果不错
看着有点长,再优化一点,让它再简介亿点
def lastStoneWeight(stones):
while len(stones) >= 2: # 将if语句的条件变成while循环的条件
stones.sort()
stones.append(stones.pop() - stones.pop()) # 将取出最大第二大石块,将两者相减,
# 添加剩余石块压缩为一句
return stones[0]
(因为pop()删除方法默认删除最后一个元素,所以pop(-1)等同于pop())
还有就是,当列表较长的时候, 使用排序运行起来是比较慢的,使用优先队列(最大,最小堆)的方法比较块一些,有兴趣可自行了解。
搬完了,下班。
题目来源:力扣