leetcode 735. 行星碰撞

给定一个整数数组 asteroids,表示在同一行的行星。

对于数组中的每一个元素,其绝对值表示行星的大小,正负表示行星的移动方向(正表示向右移动,负表示向左移动)。
每一颗行星以相同的速度移动。

找出碰撞后剩下的所有行星。碰撞规则:两个行星相互碰撞,较小的行星会爆炸。
如果两颗行星大小相同,则两颗行星都会爆炸。两颗移动方向相同的行星,永远不会发生碰撞。

示例 1:

输入: 
asteroids = [5, 10, -5]
输出: [5, 10]
解释: 
10-5 碰撞后只剩下 10510 永远不会发生碰撞。
示例 2:

输入: 
asteroids = [8, -8]
输出: []
解释: 
8-8 碰撞后,两者都发生爆炸。
示例 3:

输入: 
asteroids = [10, 2, -5]
输出: [10]
解释: 
2-5 发生碰撞后剩下 -510-5 发生碰撞后剩下 10。
示例 4:

输入: 
asteroids = [-2, -1, 1, 2]
输出: [-2, -1, 1, 2]
解释: 
-2-1 向左移动,而 12 向右移动。
由于移动方向相同的行星不会发生碰撞,所以最终没有行星发生碰撞。

方法:元素依次入栈,只有栈顶是正数,当前是负数才会碰撞。官方题解写法,用break 循环的方式,不用特判当前元素未入栈的情况了

class Solution {
    public int[] asteroidCollision(int[] asteroids) {
        int n = asteroids.length;
        Stack<Integer> stack = new Stack<>();

        for(int as : asteroids){
            collision: {
                while(!stack.isEmpty() && as < 0 && stack.peek() > 0){
                    if(stack.peek() + as < 0){ //peek被撞
                        stack.pop();
                        continue;
                    } else if(stack.peek() + as == 0){
                        stack.pop();
                    }
                    break collision;//当前元素被撞
                }
                stack.push(as);
            }
        }
        int[] res = new int[stack.size()];
        for(int i = res.length - 1; i >= 0; i--){
            res[i] = stack.pop();
        }
        return res;
    }
}

你可能感兴趣的:(leetcode,数据结构)