力扣735行星碰撞(C++实现)

详细解释都在代码注释里面了，关键在于理清楚边界条件。

class Solution {
public:
    vector<int> asteroidCollision(vector<int>& asteroids) {
        vector<int> res;
        int flag = asteroids[0]; // 记录当前栈容器内的行星走向
        int end = 0; // 用于记录最后一次碰撞的状态（是否俩质量相等的行星碰撞）
        res.push_back(asteroids[0]);
        for (int i = 1; i < asteroids.size(); ++i) {
            if (res.size() == 0) {  // 这里是为了防止下方质量相等的两颗行星碰撞后导致栈为空的情况
                res.push_back(asteroids[i]);
                continue;
            }

            if (flag * asteroids[i] > 0) { // 当前行星于栈内行星运行方向一致
                res.push_back(asteroids[i]);
                continue;
            }
            if (asteroids[i] > 0 && res[res.size() - 1] < 0) { // 如果栈内行星往左运行新加的往右运行那么永远不会相撞
                res.push_back(asteroids[i]);
                flag = asteroids[i];
                continue;
            }
            while (res.size() != 0) {
                if (res[res.size() - 1] * asteroids[i] > 0) { // 与栈顶同方向，直接进栈
                    res.push_back(asteroids[i]);
                    break;
                }
                if (abs(res[res.size() - 1]) > abs(asteroids[i])) { // 碰撞的两种情况
                    break;
                } else if (abs(res[res.size() - 1]) == abs(asteroids[i])) {
                    end = 1;
                    res.pop_back();
                    break;
                }
                res.pop_back();
            }

            if (res.size() == 0 && end != 1) { // 如果栈被弹空并且最后一次碰撞的不是两颗相等的行星
                res.push_back(asteroids[i]);
                flag = asteroids[i];
            }
            end = 0;
        }
        return res;

    }
};

由于之前我的做法圈复杂度过高，不符合代码可阅读易维护的原则，现提供一种全新解法，将模拟过程简化即可。

class Solution {
public:
    vector<int> asteroidCollision(vector<int>& asteroids) {
        vector<int> res;
        for (int i = 0; i < asteroids.size(); ++i) {
            bool alive = true;
            // 往右边走的无论什么都可以直接入栈 往左边走的满足 栈为空 或者与栈顶元素同向也可入栈
            // 否则进行碰撞检测
            while (alive && asteroids[i] < 0 && !res.empty() && res.back() > 0) {
                alive = res.back() < asteroids[i] * -1;
                if (res.back() <= asteroids[i] * -1) {
                    res.pop_back();
                }
            }
            if (alive) {
                res.push_back(asteroids[i]);
            }
        }
        return res;

    }
};