735. 行星碰撞

最新推荐文章于 2024-06-24 16:34:31 发布

原创最新推荐文章于 2024-06-24 16:34:31 发布 · 592 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

121 篇文章

订阅专栏

本文详细解析了一种基于栈结构的行星碰撞模拟算法，通过具体示例展示了如何判断行星碰撞并确定碰撞后的剩余行星。算法考虑了不同方向移动的行星以及相同大小行星碰撞的情况。

给定一个整数数组 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 向右移动。
由于移动方向相同的行星不会发生碰撞，所以最终没有行星发生碰撞。

说明:

数组 asteroids 的长度不超过 10000。
每一颗行星的大小都是非零整数，范围是 [-1000, 1000] 。

Reivew:

使用栈结构,但要注意空栈情况以及相等碰撞情况.

Code:

class Solution {
    public int[] asteroidCollision(int[] asteroids) {
        Stack<Integer> stack = new Stack<>();
        for (int temp : asteroids) {
            if (temp > 0) {
                stack.push(temp);
            } else {
                while (!stack.empty() && stack.peek() > 0 && stack.peek() < -temp) {
                    stack.pop();
                }
                if (!stack.empty() && stack.peek() == -temp) {
                    stack.pop();
                    continue;
                }
                if (stack.empty() || stack.peek() < 0) {
                    stack.push(temp);
                }
            }
        }
        int[] res = new int[stack.size()];
        for (int i = stack.size() - 1; i > -1; i--) {
            res[i] = stack.pop();
        }
        return res;
    }
}