735. 行星碰撞

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

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/asteroid-collision
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目分析：

遍历数组，每次得到的数字，要么是正数，要么是负数。
根据题意，负数是向左走，正数是向右走。
前后两次遍历到的数字，总共有四种可能，即：

1. +，+ 不会爆炸
2. +，- 爆炸
3. -，+ 不会爆炸
4. -，- 不会爆炸

从上面的分析可以看出，如果前一个数是负数的话，不可能与后一个数发生爆炸，如果前一个数是正数的话，则会爆炸，保留谁，则要看二者之间的大小关系：

• 正数+负数=0，两个数都没了
• 正数+负数<0,保留负数
• 正数+负数>0,保留正数

程序设计

用栈保存结果
遍历数组，当前遍历到item

• 如果item是整数，则保存在栈中
• 如果是负数，会不会爆炸则与栈顶元素有关。
• 步骤一
  • 如果栈是空的，则保存item，
  • 如果栈顶是负数，不爆炸，item入栈
  • 如果栈顶top是正数，爆炸，
    1）top + item > 0，则不保存item，继续下一轮的遍历
    2）top + item == 0，则不保存item，同时弹出top，继续下一轮的遍历
    3）top + item < 0, 即top没有item的绝对值大，则需要弹出top, item继续与栈顶元素比较，跳到步骤一处进行比较。

'只有正数才有可以能爆炸'
class Solution:
    def asteroidCollision(self, asteroids):
        stack = []
        flag = 0
        for item in asteroids:
            if item > 0:# 遍历到的是正数，直接压栈，等着被爆炸
                stack.append(item)
            else:# 遍历的是负数，就要判断栈顶是不是正数

                while stack: 
                    if stack[-1] > 0:
                        if stack[-1] + item < 0: # top + item < 0,需要弹出top, item继续与栈顶元素比较
                            stack.pop()


                        elif stack[-1] + item == 0:# 不保存`item`，同时弹出`top`，继续下一轮的遍历
                            stack.pop()
                            break
                        else:# `top + item > 0`，则不保存`item`，继续下一轮的遍历
                            break
                    else:# 
                        stack.append(item)# 栈顶是负数，不爆炸，`item`入栈，继续下一轮的遍历
                        break
                else:# 栈为空，压栈
                    stack.append(item)
        return stack

asteroids = [5,10,-5]
# asteroids = [8,-8]
# asteroids = [10,2,-5]
# asteroids = [-2,-1,1,2]
# asteroids = [1,-2,-2,-2]
# asteroids = [-2,5,8,-9]
# asteroids = [-2,-2,1,-2]
# # 输出：[5,10]
A = Solution()
b = A.asteroidCollision(asteroids)
print(b)