【每日一算法】行星碰撞

最新推荐文章于 2022-09-11 20:23:36 发布

视学算法

最新推荐文章于 2022-09-11 20:23:36 发布

阅读量595

点赞数

本文链接：https://blog.csdn.net/zandaoguang/article/details/100117152

版权

微信改版，加星标不迷路！

640?wx_fmt=png

每日一算法-行星碰撞

给定一个整数数组 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 向右移动。
由于移动方向相同的行星不会发生碰撞，所以最终没有行星发生碰撞。

说明:

数组 asteroids 的长度不超过 10000。
每一颗行星的大小都是非零整数，范围是 [-1000, 1000] 。

解析

思路

将asteroids中的数字通过某种算法选择性挪到新定义的res中，最后返回res。

算法(伪代码)

1.如果asteroids[i]是正数，则直接加入res.push_back(asteroids[i]);

2.如果asteroids[i]是负数，则进行以下判断：

2.1如果res为空或者res.back()为负数，则直接加入res.push_back(asteroids[i]);

2.2如果res.back（） < (-asteroids[i])，则执行res.po_back() ，于此同时，因为还不能判断pop之后的res.back（）是不是符合直接插入asteroids[i]，所以现在不能直接加入asteroids[i]，我们要执行i--，为了进一步判断能够插入asteroids[i]。

2.3如果res.back() == (-asteroids[i])，即要执行res.pop_back()。因为这里不需要插入-asteroids[i]，所以这里不用执行i--。

实现代码

class Solution {
public:
    vector<int> asteroidCollision(vector<int>& asteroids) {
        vector<int> res;
        for(int i = 0; i < asteroids.size() ; i++){
            //如果是正数，则直接插入
            if(asteroids[i] > 0){
                res.push_back(asteroids[i]);
            }
            //如果是负数，则进行判断。
            else{
                //如果是空，或者res的最后一个小于0的时候，负数才可以插入
                if(res.empty() || res.back() < 0){
                    res.push_back(asteroids[i]);
                }
                //如果res的最后一个小于|要插入的值|，则将res的最后一个退出，然后依次判断res是否符合...
                else if(res.back() < -asteroids[i]){
                        i--;
                        res.pop_back();
                }
                //如果res的最后一个==|要插入的值|，则res最后一个pop，于此同时，也不插入当前的asteroids[i]
                else if(res.back() == -asteroids[i]){
                        res.pop_back();
                }
            }
                
        }
        return res;
    }
};