735. 行星碰撞

题目

给定一个整数数组 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 <= 10⁴
• -1000 <= asteroids[i] <= 1000
• asteroids[i] != 0

题解

这题要比较相近的两个元素之间的关系，我们考虑用栈来做。
根据题意，只有栈顶元素值 > 0 并且 新元素值 < 0 时，才会发生爆炸，爆炸有三种情况：

• 值相同，一起爆炸；
• 栈顶元素值大，新元素爆炸；
• 新元素大，栈顶元素爆炸，直到栈顶元素比新元素大 或者 栈空 为止。

代码实现

func asteroidCollision(asteroids []int) []int {
	if len(asteroids) == 0 {
		return nil
	}
	var stack []int
	stack = append(stack, asteroids[0])
	var top int // 保存栈顶元素
	for i := 1; i < len(asteroids); i++ {
		if len(stack) > 0 {
			top = stack[len(stack)-1]
		} else {
			top = -1
		}
		if top < 0 || asteroids[i] > 0 { // 此时不会发生碰撞，添加到栈顶
			stack = append(stack, asteroids[i])
			continue
		}
		// 拿到栈顶元素
		for top > 0 && asteroids[i] < 0 && len(stack) > 0 { // 此时发生碰撞
			// 1. 质量相同
			if top+asteroids[i] == 0 {
				stack = stack[:len(stack)-1]
				break
			}
			// 2. 栈顶 > 新元素
			if top+asteroids[i] > 0 {
				break
			}
			// 3. 栈顶 < 新元素
			if top+asteroids[i] < 0 {
				stack = stack[:len(stack)-1]
				top = -1
				if len(stack) > 0 {
					top = stack[len(stack)-1]
				}
				if top < 0 {
					stack = append(stack, asteroids[i])
				}
			}
		}
	}
	return stack
}