给定一个整数数组 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博客
使用栈 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)。