小行星碰撞

/*
 * 思路：
 * 使用Stack存储行星。对于每个小行星，如果是负数且栈顶是正数，则表示可能发生碰撞。否则直接入栈。
 * 使用Stream生成最终结果。
 */
import java.util.Stack;
import java.util.Arrays;
 
public class AsteroidCollisionStream {
    public int[] asteroidCollision(int[] asteroids) {
        Stack<Integer> stack = new Stack<>();
        Arrays.stream(asteroids).forEach(asteroid -> {
            boolean alive = true;
            while (alive && !stack.isEmpty() && stack.peek() > 0 && asteroid < 0) {
                int top = stack.pop();
                if (top + asteroid == 0) {
                    alive = false; // both explode
                } else if (top + asteroid > 0) {
                    asteroid = top; // top survives
                } // else, asteroid survives (keep alive = true)
            }
            if (alive) stack.push(asteroid);
        });
        return stack.stream().mapToInt(i -> i).toArray();
    }
}