题目链接

栈排序

题目描述

注意点

• 对栈进行排序使最小元素位于栈顶
• 最多只能使用一个其他的临时栈存放数据
• 不得将元素复制到别的数据结构（如数组）中
• 栈中的元素数目在[0, 5000]范围内

解答思路

• 本题是要实现一个小顶堆，可以直接使用PriorityQueue
• 如果要使用栈完成该结构，则需要在入栈时对栈中元素进行排序：如果入栈的val比栈顶元素小，则直接将val加入到栈中；如果入栈的val比栈顶元素大，则需要先将栈中元素弹出并存储，直到val比栈顶元素小为止，然后将val加入到栈中，再将弹出的栈中元素按顺序添加到栈中（保证栈中的元素始终按从小到大进行排序）

代码

class SortedStack {Stack<Integer> stk;public SortedStack() {stk = new Stack<>();}public void push(int val) {List<Integer> list = new ArrayList<>();while (!stk.isEmpty() && val > stk.peek()) {list.add(stk.pop());}stk.push(val);for (int i = list.size() - 1; i >= 0; i--) {stk.push(list.get(i));}}public void pop() {if (!stk.isEmpty()) {stk.pop();}}public int peek() {return stk.isEmpty() ? -1 : stk.peek();}public boolean isEmpty() {return stk.isEmpty();}
}

关键点

• 无