LeetCode 程序员面试金典 面试题 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]

用一个辅助栈，维持data栈的有序状态，（每次弹出一些元素到辅助栈，方便新的元素插入到所需位置，然后再从辅助栈中压回去）

class SortedStack {
    public Stack<Integer> data;
    public Stack<Integer> help;
    public SortedStack() {
        data = new Stack<>();
        help = new Stack<>();
    }

    public void push(int val) {
        if(data.empty() || data.peek() >= val) {
            data.push(val);
            return ;
        }
        while (!data.empty() &&data.peek() < val) {
            help.push(data.pop());
        }
        data.push(val);
        while(!help.empty()) {
            data.push(help.pop());
        }
    }

    public void pop() {
        
        if(!data.empty())
        data.pop();
    }

    public int peek() {
        
        if(!data.empty())
        return data.peek();
        return -1;
    }

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