题目
给定一个整数数组 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 <= 104
-1000 <= asteroids[i] <= 1000
asteroids[i] != 0
代码
package dayLeetCode;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.Stack;
public class dayleetcode735 {
// 模拟 栈
// 相当于两伙人打架,正数一波,负数一波,找出最后剩下的人,下面将正负数称为要打架的人
// 我们将正数存入栈中,每当一个负数到来的时候,就取离他最近的正数来应对,负数赢了就进入栈中,负数输了就消失,同归于尽两个都消失
public int[] asteroidCollision(int[] asteroids) {
Stack<Integer> stack = new Stack<>();
for (int aster : asteroids){
// 标识行星是否存活
boolean flag = true;
while (flag && !stack.isEmpty() && stack.peek() > 0 && aster < 0){
int tmp1 = Math.abs(stack.peek()), tmp2 = Math.abs(aster);
// 正数被消灭
if (tmp1 < tmp2){
stack.pop();
}
// 负数被消灭
else if (tmp1 > tmp2){
flag = false;
} else {
// 同归于尽
stack.pop();
flag = false;
}
}
if (flag){
stack.push(aster);
}
}
int len = stack.size();
int[] ans = new int[len];
for (int i = len - 1; i >= 0; i--){
ans[i] = stack.pop();
}
return ans;
}
public static void main(String[] args) {
dayleetcode735 obj = new dayleetcode735();
for (int num : obj.asteroidCollision(new int[]{5, 10, -5})) {
System.out.println(num);
}
}
}