Leetcode刷题记录 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 。

class Solution {
public:
    vector<int> asteroidCollision(vector<int>& asteroids) {
        stack<int> s; //用于保存不碰撞的元素
        s.push(asteroids[0]);
        for(int i=1;i<asteroids.size();i++){
            //一直要注意这几个边界条件
            if(s.empty() || (asteroids[i]>0 && s.top()>0) || s.top()<0){
                s.push(asteroids[i]);
            }
            else{
                int x=asteroids[i];
                /*自己做的时候一直会报一个空指针的错误，
                原因是此处要把s.empty()放在第一个，
                不然在判断s.top()的时候就会报空指针的错误
                */
                while(!s.empty() && x<0 && s.top()>0){
                    if(abs(x)<s.top()){
                        x=s.top();
                        s.pop();
                        break;
                    }
                    else if(abs(x)==s.top()){
                        x=0; //0代表后续两个都不会进栈
                        s.pop();
                        break;
                    }
                    else{
                        s.pop();
                    }
                }
                if(x!=0){
                    s.push(x);
                }
            }
        }
        int n=s.size();
        vector<int> result(n);
        for(int i=n-1;i>=0;i--){
            result[i]=s.top();
            s.pop();
        }
        return result;
    }
};