给定一个整数数组 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] 。
思路:用栈模拟,碰到值大于0的无脑加入栈,如果当前值小于0,判断栈中是否有向右移动的更大的行星,没有的的话将负值加入即可。
class Solution {
public int[] asteroidCollision(int[] asteroids) {
Stack<Integer> st=new Stack<>();
for(int i=0;i<asteroids.length;i++) {
if(asteroids[i]>0 || st.isEmpty() || asteroids[st.peek()]<0)
st.add(i);
else {
boolean flag=false;
while(!st.isEmpty() && asteroids[st.peek()]>0 && asteroids[st.peek()]<=-asteroids[i]) {
if(asteroids[st.peek()]==-asteroids[i])
flag=true;
st.pop();
if(flag) break;
}
if(!st.isEmpty() && asteroids[st.peek()]>0 && asteroids[st.peek()]>-asteroids[i] || flag)
continue;
st.add(i);
}
}
int len=0;
int[] ans=new int[st.size()];
while(!st.isEmpty())
ans[len++]=st.pop();
Arrays.parallelSort(ans);
for(int i=0;i<ans.length;i++)
ans[i]=asteroids[ans[i]];
return ans;
}
}