面试题 03.05. 栈排序

题目

栈排序。 编写程序，对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据，但不得将元素复制到别的数据结构（如数组）中。该栈支持如下操作：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]范围内。

---

解题思路：

1.使用两个栈搞定

思路：

• 思路非常简单，将值存储stack1之前先判断栈顶元素是否大于push的值
• 如果大于，则将其存入第二个stack2中，然后判断下一个的值。。。以此类推直到stack1空或者找到小于push的值
• 然后将值push入栈stack1，之后将刚刚push到stack2中的值再倒回stack1中。

代码：

class SortedStack {
    private Stack<Integer> stack1;
    private Stack<Integer> stack2;

    public SortedStack() {
        this.stack1 = new Stack<>();
        this.stack2 = new Stack<>();
    }

    public void push(int val) {
        if (stack1.isEmpty()) {
            stack1.push(val);
        } else {
            while (!stack1.isEmpty() && val>stack1.peek()){
                stack2.push(stack1.pop());
            }
            stack1.push(val);
            while (!stack2.isEmpty()){
                stack1.push(stack2.pop());
            }
        }
    }

    public void pop() {
        if (!stack1.isEmpty()){
            stack1.pop();
        }
    }

    public int peek() {
        if (!stack1.isEmpty()){
            return stack1.peek();
        }
        return -1;
    }

    public boolean isEmpty() {
        return stack1.isEmpty();
    }
}

性能：
push方法因为有倾倒行文，所以时间复杂度是O(n)，其他为O（1）,空间复杂度也为O(n)