给定一个整数数组 asteroids,表示在同一行的行星。 对于数组中的每一个元素,其绝对值表示行星的大小,正负表示行星的移动方向(正表示向右移动,负表示向左移动)。每一颗行星以相同的速度移动。 找出碰撞后剩下的所有行星。碰撞规则:两个行星相互碰撞,较小的行星会爆炸。如果两颗行星大小相同,则两颗行星都会爆炸。两颗移动方向相同的行星,永远不会发生碰撞。
输入:asteroids = [10,2,-5] 输出:[10] 解释:2 和 -5 发生碰撞后剩下 -5 。10 和 -5 发生碰撞后剩下 10 。
class Solution {
public int[] asteroidCollision(int[] a) {
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < a.length; i++) {
if (stack.empty()) {
stack.push(a[i]); // 初始化栈
} else if ((stack.peek() < 0 && a[i] < 0) || (stack.peek() > 0 && a[i] > 0) || (stack.peek() < 0 && a[i] > 0))
stack.push(a[i]);//当栈顶位置元素和数组下个正负一致
else if (stack.peek() > 0 && a[i] < 0)//加进来负数
{
if (-a[i] < stack.peek())//绝对值偏“小”
continue;//2.1
else if (-a[i] == stack.peek())//相等
{
stack.pop();//2.2
continue;
} else//绝对值偏“大”
{
while (!stack.isEmpty() && stack.peek() > 0 && stack.peek() < (-a[i])) {
stack.pop();//2.3
}
if(stack.empty()||stack.peek()<0)
stack.push(a[i]);//插入数组元素
else if (!stack.empty() && -a[i] == stack.peek()) {
stack.pop();
continue;
}
}
}
}
int[] temp = new int[stack.size()];
for (int i = 0; i < temp.length; i++) {//将栈转换为数组
if (stack.get(i) != 0) {
temp[i] = stack.get(i);
}
}
return temp;
}
}