栈排序(LeetCode)

本文介绍了LeetCode中栈排序的问题,通过维护一个有序栈s1来确保最小元素始终在栈顶。push操作时,若新元素大于等于s1栈顶元素则直接入s1,否则将s1元素弹出至辅助栈s2,再将新元素入s1,最后将s2元素回填s1。pop操作直接返回栈顶元素,因为栈顶总是最小值。
摘要由CSDN通过智能技术生成

原题来自LeetCode:链接

题目描述

栈排序。 编写程序,对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push、pop、peek 和 isEmpty。当栈为空时,peek 返回 -1。

示例1:

 输入:
["SortedStack", "push", "push", "peek", "pop", "peek"]
[[], [1], [2], [], [], []]
 输出:
[null,null,null,1,null,2]

示例2:

 输入: 
["SortedStack", "pop", "pop", "push", "pop", "isEmpty"]
[[], [], [], [1], [], []]
 输出:
[null,null,null,null,null,true]

说明:

栈中的元素数目在[0, 5000]范围内。

思路

一开始以为只能用一个栈,后来看了评论发现题目说可以使用一个临时栈,加上存储数据所用的栈,就是可以用两个栈的意思。

可以用两个栈就很简单了,我们始终维护一个有序的栈s1,这个栈自顶向下的元素依次增大,即栈顶最小。

当我们需要push一个值val时,首先判断val是否小于等于s1的栈顶:

  • 如果是,那就直接push进s1即可
  • 如果不是,那就不断取出s1的元素放入辅助栈s2中,直到s1的栈顶元素大于等于val,然后将val值push进s1,最后将s2里全部的值放回s1里面。

pop时直接pop出栈顶即可,因为我们已经保证了栈顶是最小的。

class SortedStack {
   
public:
    stack<int> s1;	//存储数据的栈
    stack<int
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值