分题目:输入一个表示小行星的数组,数组中每个数字的绝对值表示小行星的大小,数字的正负号表示小行星运动的方向,正号表示向右飞行,负号表示向左飞行。如果2颗小行星相撞,那么体积小的行星将会爆炸最终消失,体积大的小行星不受影响。如果相撞的两颗小行星大小相等,那么它们会爆炸消失。飞行方向相同的小行星永远不会相撞。求最终剩下的小行星。
例如,有6颗小行星[4,5,-6,4,8,-5],它们相撞之后最终剩下3颗小行星[-6,4,8]。
#include <vector>
#include <stack>
#include <iostream>
using namespace std;
void asteroidCollision(vector<int> asteroids)
{
stack<int> stack;
for (vector<int>::iterator it = asteroids.begin(); it < asteroids.end(); it++)
{
while (!stack.empty() && stack.top() > 0 && stack.top() < -(*it)) {
stack.pop();//出栈的条件
}
if (!stack.empty() && (*it) < 0 && stack.top() == -(*it)) {
stack.pop();//相等就出栈
}
else if ((*it) > 0 || stack.empty() || stack.top() < 0) {
stack.push(*it);//入栈
}
}
cout << "size = " << stack.size() << ", data : " ;
while (!stack.empty()) {
cout << " " << stack.top();
stack.pop();
}
cout << endl;
}
分析
简单来说就3个条件:
- 相等一起消失;
- 大于0直接入栈;
- 小于0时,则需要出栈比较大小;
程序妙就妙在: while (!stack.empty() && stack.top() > 0 && stack.top() < -(*it))
如果你搞懂了这一句,就好理解了!