剑指 Offer II 037. 小行星碰撞(栈)

文章目录


题目

给定一个整数数组 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):

  1. 栈为空时,入栈
  2. 当栈顶元素为,当前遍历元素为,入栈(依据上面 ab行星碰撞规则,背道而驰)
  3. 当前遍历元素为正数时,入栈(依据上面ab行星碰撞规则,b为正数时,不会碰撞)

碰撞时的出栈情况:

  1. 当 栈顶元素 小于 当前元素,栈顶元素弹出。(继续比较下一个栈顶元素,至到栈空)
  2. 当 栈顶元素 等于 当前元素,栈顶弹出,遍历下一个元素
  3. 当 栈顶元素 大于 当前元素,遍历下一个元素(栈存储的是答案,直接遍历下一个元素相当于当前元素跳过也就是爆炸)

最后把栈元素弹出保存至结果数组,输出。

代码如下:

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/

侵删

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值