leetcode 735. 行星碰撞

1. 题目链接 https://leetcode-cn.com/problems/asteroid-collision/comments/

2. 题目描述

   1. 给定一个整数数组 asteroids，表示在同一行的行星。

      对于数组中的每一个元素，其绝对值表示行星的大小，正负表示行星的移动方向（正表示向右移动，负表示向左移动）。每一颗行星以相同的速度移动。

      找出碰撞后剩下的所有行星。碰撞规则：两个行星相互碰撞，较小的行星会爆炸。如果两颗行星大小相同，则两颗行星都会爆炸。两颗移动方向相同的行星，永远不会发生碰撞。

   2. 示例 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 向右移动。
      由于移动方向相同的行星不会发生碰撞，所以最终没有行星发生碰撞。
      

3. 解题思路

   1. 栈模拟，维护一个栈，栈中的元素为保留下来的星球，遍历数组，分为以下几种情况
      1. 当前元素 > 0 直接入栈，因为栈顶元素无论正负都不会撞到当前元素
      2. 当前元素 < 0 在栈顶元素为正且绝对值小于当前元素的情况下出栈，出栈完毕后，如果栈顶元素为正直接入栈，如果为负且绝对值与当前元素相等，那么栈顶出栈且不入栈，若大于当前元素，也不入栈。

4. 代码

   1. python

      class Solution:
          def asteroidCollision(self, asteroids):
              s = []
              for i in asteroids:
                  if not s or i > 0 or s[-1] < 0:
                      s.append(i)
                  else:
                      while s and s[-1] > 0 and s[-1] < -i:
                          s.pop()
                      if s and s[-1] == -i:
                          s.pop()
                      elif not s or s[-1] < 0:
                          s.append(i)
      
              return s