中等栈数组
PROBLEM / 研究生阶段
735. 小行星碰撞
给定一个整数数组 asteroids,表示在同一行的小行星。数组中小行星的索引表示它们在空间中的相对位置。
对于数组中的每一个元素,其绝对值表示小行星的大小,正负表示小行星的移动方向(正表示向右移动,负表示向左移动)。每一颗小行星以相同的速度移动。
找出碰撞后剩下的所有小行星。碰撞规则:两个小行星相互碰撞,较小的小行星会爆炸。如果两颗小行星大小相同,则两颗小行星都会爆炸。两颗移动方向相同的小行星,永远不会发生碰撞。
示例 1:
输入: asteroids = 5,10,-5输出:5,10解释: 10 和 -5 碰撞后只剩下 10 。 5 和 10 永远不会发生碰撞。
示例 2:
输入: asteroids = 8,-8输出:解释: 8 和 -8 碰撞后,两者都发生爆炸。
示例 3:
输入: asteroids = 10,2,-5输出:10解释: 2 和 -5 发生碰撞后剩下 -5 。10 和 -5 发生碰撞后剩下 10 。
示例 4:
输入: asteroids = 3,5,-6,2,-1,4输出:-6,2,4解释: 小行星 -6 使小行星 3 和 5 爆炸,然后继续向左移动。在另一边,小行星 2 使小行星 -1 爆炸,然> 后继续向右移动,没有碰撞小行星 4。
提示:
2 <= asteroids.length <= 104-1000 <= asteroids[i] <= 1000asteroids[i] != 0
参考解法
下面保留的是原始练习仓库中的个人作答,可在右侧工作台中独立重写,再按需揭示对照。
python
class Solution(object):
def asteroidCollision(self, asteroids):
stack = []
for ast in asteroids:
destoryed = False # 标记当前行星是否被撞毁(默认False)
# 栈不为空/栈顶向右/当前向左 发生碰撞
while stack and stack[-1] > 0 and ast < 0 and not destoryed:
if abs(ast) > abs(stack[-1]):
stack.pop()
elif abs(ast) < abs(stack[-1]):
destoryed = True
else:
stack.pop()
destoryed = True
if not destoryed:
stack.append(ast)
return stack
"""
:type asteroids: List[int]
:rtype: List[int]
"""
YOUR SOLUTION735. 小行星碰撞