735. 行星碰撞 https://leetcode-cn.com/problems/asteroid-collision/
给定一个整数数组 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]
。
思路1:使用0标记处理过的。
class Solution {
public:
vector<int> asteroidCollision(vector<int>& asteroids) {
bool isexchanged=false;
int n=asteroids.size();
int istart=0;
do {
isexchanged=false;
istart=0;
for(int i=istart; i<n; i++) {
if(asteroids[i]<=0) continue;
int k=1;
while(i+k<n&&asteroids[i+k]==0) k++;
if(asteroids[i+k]>0) {
i=i+k-1;
continue;
}
if(i+k<n) {
if(abs(asteroids[i])>abs(asteroids[i+k]))
asteroids[i+k]=0;
else if(abs(asteroids[i])<abs(asteroids[i+k]))
asteroids[i]=0;
else {
asteroids[i]=0;
asteroids[i+k]=0;
}
isexchanged=true;
}
istart=i;
}
} while(isexchanged);
vector<int> t;
for(int i=0; i<n; i++)
if(asteroids[i])
t.push_back(asteroids[i]);
return t;
}
};
解法2:使用堆栈
class Solution {
public:
vector<int> asteroidCollision(vector<int>& asteroids) {
vector<int> ans;
if(asteroids.empty())
return ans;
stack<int> st;
int len = asteroids.size();
for(int i = 0;i < len;i++){
if(st.empty() || (st.top() < 0 && asteroids[i] > 0) || (st.top() * asteroids[i] > 0)) //如果栈为空、当前行星和栈顶行星相背或同向运行,就入栈
st.push(asteroids[i]);
else if(abs(st.top()) == abs(asteroids[i])) //如果当前行星和栈顶行星相向而行且大小相同,同时爆炸
st.pop();
else if(abs(st.top()) < abs(asteroids[i])){ //如果当前行星和栈顶行星相向而行且大小相同,栈顶小于当前
bool flag = true; //flag标识当前行星是否会爆炸
while(!st.empty() && st.top() > 0 && abs(st.top()) <= abs(asteroids[i])){
if(abs(st.top()) == abs(asteroids[i])){
flag = false;
st.pop();
break;
}
st.pop();
}
if(flag && (st.empty() || (st.top() * asteroids[i] > 0)))
st.push(asteroids[i]);
}
}
len = st.size();
ans.resize(len);
for(int i = len - 1;i >= 0;i--){
ans[i] = st.top();
st.pop();
}
return ans;
}
};