LeetCode 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 。

提示：

• 2 <= asteroids.length <= 10^4
• -1000 <= asteroids[i] <= 1000
• asteroids[i] != 0

---

提示 1

Say a row of asteroids is stable. What happens when a new asteroid is added on the right?

解法1：栈

相似题型：LeetCode 2211. 统计道路上的碰撞次数-CSDN博客

LeetCode 2751. 机器人碰撞-CSDN博客

使用栈 stack 模拟行星碰撞，从左往右遍历行星数组 asteroids，当我们遍历到行星 aster 时，使用变量 alive 记录行星 aster 是否还存在（即未爆炸）。

当行星 aster 存在且 aster<0，栈顶元素非空且大于 0 时，说明两个行星相互向对方移动：

如果栈顶元素大于等于 −aster，则行星 aster 发生爆炸，将 alive 置为 false；

如果栈顶元素小于等于 −aster，则栈顶元素表示的行星发生爆炸，执行出栈操作。

重复以上判断直到不满足条件，如果最后 alive 为真，说明行星 aster 不会爆炸，则将 aster 入栈。

为了代码简洁性，我们使用变长数组模拟栈。

Java版：

class Solution {
    public int[] asteroidCollision(int[] asteroids) {
        Stack<Integer> stack = new Stack<>();
        for (int aster : asteroids) {
            boolean alive = true;
            while (alive && aster < 0 && !stack.isEmpty() && stack.peek() > 0) {
                alive = -aster > stack.peek();
                if (stack.peek() <= -aster) {
                    stack.pop();
                }
            }
            if (alive) {
                stack.push(aster);
            }
        }
        return stack.stream().mapToInt(Integer::valueOf).toArray();
    }
}

另一种写法：

class Solution {
    public int[] asteroidCollision(int[] asteroids) {
        Deque<Integer> stack = new ArrayDeque<>();
        for (int aster : asteroids) {
            boolean alive = true;
            while (alive && aster < 0 && !stack.isEmpty() && stack.peek() > 0) {
                alive = -aster > stack.peek();
                if (stack.peek() <= -aster) {
                    stack.pop();
                }
            }
            if (alive) {
                stack.push(aster);
            }
        }
        int n = stack.size();
        int[] ans = new int[n];
        for (int i = n - 1; i >= 0; i--) {
            ans[i] = stack.pop();
        }
        return ans;
    }
}

 

Python3版：

class Solution:
    def asteroidCollision(self, asteroids: List[int]) -> List[int]:
        stack = []
        for aster in asteroids:
            alive = True
            while alive and stack and aster < 0 and stack[-1] > 0:
                alive = -aster > stack[-1]
                if stack[-1] <= -aster:
                    stack.pop()
            if alive:
                stack.append(aster)
        return stack        
                

复杂度分析

• 时间复杂度：O(n)，其中 n 为数组 asteroids 的大小。出入栈次数均不超过 n 次。
• 空间复杂度：O(n)。