栈排序(使使最小元素位于栈顶)

题目:

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

示例:

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

解题思路:

创建一个辅助栈s2,当要将val入栈s1时,判断s1的元素是否小于val,只要小于,就将元素放入s2栈中,暂时保存起来,直到找到比val大的就停止,然后将val入栈s1,再将刚刚在s2中保存的所有元素依次入栈s1,这样就能始终保持s1中的元素自底向上是按从大到小的顺序存放的。

eg.

当前s1栈的元素:5,3,2,1  ,val=4

1.通过比较,先将1,2,3,依次出栈并存放到s2中,此时 s2: 1,2,3   s1:5

2.将val入栈。   此时:s1:5,4

3.再将元素归还给s1。此时:s1:5,4,3,2,1   s2:null

 源代码如下:

class SortedStack {
public:
    stack s1;//原始栈
    stack s2;//辅助栈,该栈用来保存s1栈中所有比val小的值
    SortedStack() {

    }
    //元素入栈
    void push(int val) {
        //如果栈为空,直接入栈就可以了,此时该元素就是栈中最小元素
        if(s1.empty())
        {
            s1.push(val);
            return ;
        }
        //否则,要判断s1栈顶元素与val的大小 
        //如果s1的栈顶元素小于此时要入栈的元素,就要保存这个小元素
        while(!s1.empty()&&s1.top()

你可能感兴趣的:(数据结构,leetcode,c++,算法)