题目
给定一个整数数组 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向右移动。 由于移动方向相同的行星不会发生碰撞,所以最终没有行星发生碰撞。
提示:
- 2 <= asteroids.length <= 104
- -1000 <= asteroids[i] <= 1000
- asteroids[i] != 0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/XagZNi
一、栈
解题思路:
我们需要明确出什么情况下两颗行星碰撞:
假设有左右 a b两颗行星
- 当 ab 都为正,不会碰撞
- 当 ab 都为负,不会碰撞
- 当 a为负,b为正,不会碰撞(背道而驰 <–a b–>)
- 当 a为正,b为负,碰撞(a–> <–b)
当碰撞时,出现的情况
- 当 a<|b| ,a爆炸,b将 继续 往左边移动
- 当 a=|b| ,ab都爆炸
- 当 a>|b| ,b爆炸,a将继续往右边移动
现在情况分析完了,思路一下子就清晰了
咱们定义一个存储答案的栈,正向遍历行星数组。
入栈条件(可把栈顶元素看为a,当前遍历元素看为b):
- 当栈为空时,入栈
- 当栈顶元素为负,当前遍历元素为正,入栈(依据上面 ab行星碰撞规则,背道而驰)
- 当 当前遍历元素为正数时,入栈(依据上面ab行星碰撞规则,b为正数时,不会碰撞)
碰撞时的出栈情况:
- 当 栈顶元素 小于 当前元素,栈顶元素弹出。(继续比较下一个栈顶元素,至到栈空)
- 当 栈顶元素 等于 当前元素,栈顶弹出,遍历下一个元素
- 当 栈顶元素 大于 当前元素,遍历下一个元素(栈存储的是答案,直接遍历下一个元素相当于当前元素跳过也就是爆炸)
最后把栈元素弹出保存至结果数组,输出。
代码如下:
class Solution {
public int[] asteroidCollision(int[] asteroids) {
Deque<Integer> eq = new LinkedList<>();
int index = 0;
while(index<asteroids.length){
if(eq.isEmpty()||eq.peek()<0||asteroids[index]>0) eq.push(asteroids[index]);
else if(eq.peek()<=-asteroids[index]){
// = 情况不处理,后面会执行index++,相当于把当前asteroids[index]碰撞了
//而 < 则把栈顶弹出,continue 接着新一轮比较
if(eq.pop()<-asteroids[index]) continue;
}
index++;
}
int[] res = new int[eq.size()];
for(int i=res.length-1;i>=0;i--){
res[i] = eq.pop();
}
return res;
}
}
加深栈方法使用:
剑指 Offer II 038. 每日温度(单调栈)
剑指 Offer II 036. 后缀表达式(栈)
参考qingfengpython 解析
链接:https://leetcode-cn.com/problems/XagZNi/solution/shua-chuan-jian-zhi-offer-day17-zhan-i-0-5yho/
侵删