【LeetCode 75】第二十五题(735)行星碰撞

目录

题目:

示例:

分析:

代码+运行结果:


题目:

示例:

分析:

给一个数组,数组里的元素表示行星,元素的符号决定行星运动的方向,元素的绝对值决定行星的大小,当行星发生碰撞,较小的行星会爆炸(消失),返回爆炸结束后的行星数组。

这种相邻元素发生操作的,以及有一定顺序要求的题目,都是可以用栈来做的.

由于题目要求返回vector,因此我们就用vector来模拟栈的操作,使用 *(vector.end()-1) 来获取栈顶元素,使用vector.pop_back()来弹出栈顶元素.

首先我们遍历小行星群(原数组),然后再进入判断,如果栈为空,那么不会发生碰撞,不用进一步判断,因此直接入栈.

如果小行星和栈顶的行星的同一个方向的(符号相同),那么也不会发生碰撞,也直接入栈.

然后如果小行星和栈顶行星方向不一样也不一定会发生碰撞,如果栈顶行星往左走(符号为-),而小行星往右走(符号为+),那么即使它们运动方向不一样也是不会发生碰撞的.

如果小行星左走,栈顶行星右走,那么它们就会发生碰撞,这时候我们需要进一步判断.

如果小行星更大,那么栈顶行星爆炸,我们将其出栈,并且由于可能新来的小行星会和新的栈顶行星接着发生碰撞,因此我们需要while循环判断.

如果栈顶行星更大,那么小行星爆炸,我们不用操作.

如果它们一样大,那么一起爆炸,将栈顶元素弹出,然后退出判断循环.

代码+运行结果:

cpp 复制代码
class Solution {
public:
    vector<int> asteroidCollision(vector<int>& asteroids) {
        vector<int>stack;
        for(int &a:asteroids){
            //判断栈是否为空,并且判断栈顶元素是否和新来的行星是同号
            if(stack.empty()||*(stack.end()-1)*a>0) stack.push_back(a);
            else{
                //如果栈顶元素和新来行星不同号,则可能发生碰撞(左正右负发生碰撞,左负右正不发生碰撞)
                while(!stack.empty()){
                    int temp=*(stack.end()-1);
                    if(a<0 && temp>0){  //如果是左正右负的情况则发生碰撞.
                        //如果栈顶元素比新来行星更大,那么新来的行星会爆炸,退出循环
                        if(temp>-a) break;  
                        //如果相同,两个一起爆炸,弹出栈顶元素,退出循环
                        else if(a==-temp){  
                            stack.pop_back();
                            break;
                        }
                        //如果新行星更大,那么栈顶爆炸(弹出)
                        else{
                            stack.pop_back();
                            //再做个判断,如果栈顶弹出后为空了,还需要把新行星入栈,然后退出循环.
                            if(stack.empty()){
                                stack.push_back(a);
                                break;
                          }  
                        }
                    }else{  //反之左负右正不发生碰撞,退出循环.
                        stack.push_back(a);
                        break;
                    }
                }
            }
        }
        return stack;
    }
};
相关推荐
Wilber的技术分享10 分钟前
【机器学习实战笔记 14】集成学习:XGBoost算法(一) 原理简介与快速应用
人工智能·笔记·算法·随机森林·机器学习·集成学习·xgboost
Tanecious.23 分钟前
LeetCode 876. 链表的中间结点
算法·leetcode·链表
Wo3Shi4七32 分钟前
哈希冲突
数据结构·算法·go
呆呆的小鳄鱼1 小时前
cin,cin.get()等异同点[面试题系列]
java·算法·面试
Touper.1 小时前
JavaSE -- 泛型详细介绍
java·开发语言·算法
sun0077001 小时前
std::forward作用
开发语言·c++·算法
JoernLee1 小时前
机器学习算法:支持向量机SVM
人工智能·算法·机器学习
V我五十买鸡腿2 小时前
顺序栈和链式栈
c语言·数据结构·笔记·算法
海棠玛卡2 小时前
C/C++内存管理
c++
我爱一条柴ya2 小时前
【AI大模型】线性回归:经典算法的深度解析与实战指南
人工智能·python·算法·ai·ai编程