LeetCode 735. 行星碰撞 题解 C/C++

主要思路

首先想到要用栈，这里用动态数组模拟栈，
栈为空或者前面的行星都是向左移动，则后面的行星不管向哪边移动都不会与前面的相撞,直接入栈即可
如果前面的行星向右移动，后面的也向右移动也不会相撞，直接入栈即可
只有前面的行星向右移动，后面的向左移动才会相撞，之后循环比较当前行星的质量与栈顶行星的质量，如果绝对值相等则双方都销毁，栈顶出栈，如果当前行星的质量大，则栈顶出栈，当前行星继续与栈顶行星比较，如果栈顶质量大，则当前行星爆炸，跳出循环，继续遍历后续元素。

class Solution {
public:
	vector<int> asteroidCollision(vector<int>& asteroids) {
		vector<int> stk;
		int len = asteroids.size();
		for(int i = 0;i<len;i++) {
			//栈为空或者前面的行星都是向左移动，则后面的行星不管向哪边移动都不会相撞,直接入栈即可
			if(stk.empty()||stk.back()<0){
				stk.push_back(asteroids[i]);
			}
			else if(stk.back()>0&&asteroids[i]>0) {//如果前面的行星向右移动，后面的也向右移动也不会相撞直接入栈即可
				stk.push_back(asteroids[i]);
			}
			else {//只有前面的行星向右移动，后面的向左移动才会相撞
				int tmp = asteroids[i];//暂存当前行星的大小
				int flag = 1;
				while(!stk.empty()&&stk.back()>0) {
					if((stk.back()+tmp)==0) {
						stk.pop_back();
						flag = 0;
						break;
					}
					else if((stk.back()+tmp)<0) {
						stk.pop_back();
						continue;
					}
					else {
						flag = 0;
						break;
					}
				}
				if(flag)stk.push_back(tmp);//判断当前行星是否入栈
			}
		}
		return stk;
	}
};