Python leetcode LCR 037 行星碰撞,单调栈的应用,进栈和出栈代码实战

1 leetcode LCR 037 行星碰撞

给定一个整数数组 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 = [-2,-1,1,2]
输出:[-2,-1,1,2]
解释-2 和 -1 向左移动,而 1 和 2 向右移动。 由于移动方向相同的行星不会发生碰撞,所以最终没有行星发生碰撞。 

提示:

  • 2 <= asteroids.length <= 104
  • -1000 <= asteroids[i] <= 1000
  • asteroids[i] != 0

2 题解代码

1.建立一个栈,储存没有发生爆炸的行星,python用列表或者栈包代替

2.入栈情况:栈为空,栈顶元素小于0(即向左运动),当前元素大于0(即向右运动),满足一个即可进栈;

3.出栈:不满足以上三种情况,并且栈顶元素的绝对值小于等于当前元素的绝对值,继续讨论,当弹出的元素小于当前元素abs时,当前元素不变(position不变),注意continue,继续讨论栈顶元素和当前元素;否则,当前元素爆炸,即position += 1

代码

class Solution:
    def asteroidCollision(self, asteroids: List[int]) -> List[int]:
        stack, position = [], 0
        while position < len(asteroids):
            if not stack or stack[-1] < 0 or asteroids[position] > 0:
                stack.append(asteroids[position])
            elif stack[-1] <= -asteroids[position]:
                if stack.pop() < -asteroids[position]:
                    continue
            position += 1
        return stack

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

医学小达人

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值