栈排序【大厂算法面试题】

题目描述

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

解题思路

我们用两个栈，一个原始栈用来存储全部数据，另一个临时栈用来模拟插入排序后移数组的操作，把插入的数据放在合适的位置后，在将临时栈的数据再放回原始栈。

实现代码

class SortedStack {

    private Stack<Integer> stack;
    private Stack<Integer> tmp;
    public SortedStack() {
        stack = new Stack<>();
        tmp = new Stack<>();
    }
    
    public void push(int val) {
        while(!stack.isEmpty() && stack.peek() < val) {
            tmp.push(stack.pop());
        }
        stack.push(val);
        while(!tmp.isEmpty()) {
            stack.push(tmp.pop());
        }
    }
    
    public void pop() {
        if(!stack.isEmpty()) {
            stack.pop();
        }
    }
    
    public int peek() {
        if(!stack.isEmpty()) {
            return stack.peek();
        }
        return -1;
    }
    
    public boolean isEmpty() {
        return stack.isEmpty();
    }
}

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-of-stacks-lcci