栈排序。 编写程序,对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push
、pop
、peek
和isEmpty
。当栈为空时,peek
返回-1
。
输入:
["SortedStack", "push", "push", "peek", "pop", "peek"]
[[], [1], [2], [], [], []]
输出:
[null,null,null,1,null,2]
输入:
["SortedStack", "pop", "pop", "push", "pop", "isEmpty"]
[[], [], [], [1], [], []]
输出:
[null,null,null,null,null,true]
var SortedStack = function() {
this.stack = [];
this.auxStack = [];
};
/**
* @param {number} val
* @return {void}
*/
SortedStack.prototype.push = function(val) {
if(this.stack.length){
while(true){
var topValue = this.stack.pop();
if(topValue > val || topValue === undefined){
if(topValue !== undefined) this.stack.push(topValue);
this.stack.push(val);
while(this.auxStack.length) this.stack.push(this.auxStack.pop());
break;
}else{
this.auxStack.push(topValue);
}
}
}else{
this.stack.push(val);
}
};
/**
* @return {void}
*/
SortedStack.prototype.pop = function() {
return this.stack.pop();
};
/**
* @return {number}
*/
SortedStack.prototype.peek = function() {
if (this.stack.length === 0) return -1;
return this.stack[this.stack.length - 1];
};
/**
* @return {boolean}
*/
SortedStack.prototype.isEmpty = function() {
return this.stack.length === 0;
};
/**
* Your SortedStack object will be instantiated and called as such:
* var obj = new SortedStack()
* obj.push(val)
* obj.pop()
* var param_3 = obj.peek()
* var param_4 = obj.isEmpty()
*/
本题有一些限制条件,必须使用栈结构进行排序,且最多只能使用一个其他的临时栈存放数据,在调用方式上是使用new
实例化构造函数,那么在SortedStack
构造函数上拓展原型以供调用,首先在构造函数中定义两个栈数组,分别为排序的主栈以及辅助栈,这两个栈数组只调用其相关的栈操作方法。在入栈的push
操作时就对值进行排序,如果栈中没有值,那么就直接入栈,如果存在值则首先出栈栈顶值,判断该值与入栈的值的大小,如果栈顶的值大于入栈的值或者主栈中已经出栈完毕,那么首先判断出栈的值是否有效,有效则就将出栈的值首先入栈,然后将入栈的值进栈,然后将辅助栈中的值依次出栈并压入主栈,直到辅助栈空,然后结束循环,如果栈顶的值小于出栈的值且主栈并未出栈完毕,那么将该出栈的值压入辅助栈。由于主栈在入栈时就已经有序,那么出栈的pop
操作直接调用pop()
方法即可。对于peek
操作则首先判断栈长度,如果长度为0
则返回-1
,否则返回栈顶元素的值。对于isEmpty
操作则直接判断主栈长度是否为0
即可。
https://github.com/WindrunnerMax/EveryDay
https://leetcode-cn.com/problems/sort-of-stacks-lcci