题目
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路
JS里用数组可以模拟栈:push pop,也可以模拟队列push,shift
数据存放在其中一个栈里面,push操作直接push,pop操作,先把有数据的栈全部pop出来,并push到另一个数组里面,这样直接pop,就是队列的头。
代码
/*用两个数组来模拟两个栈*/
function queue() {
this.stack1=[];
this.stack2=[];
this.push=function(node) {
while(this.stack2.length!==0){
this.stack1.push(this.stack2.pop());
}
this.stack1.push(node);
}
this.pop=function () {
while(this.stack1.length!==0){
this.stack2.push(this.stack1.pop());
}
return this.stack2.pop();
}
}
扩展:两个队列实现一个栈
push的时候,数据在哪个队列就push进哪个队列;
pop的时候,把n-1个shift掉,最后一个shift掉并且返回就可以。
/*两个队列实现一个栈*/
function stack(){
this.q1=[];
this.q2=[];
this.push=function (ele) {
if(this.q1.length===0){
this.q2.push(ele);
}else{
this.q1.push(ele);
}
}
this.pop=function () {
if(this.q1.length===0){
while(this.q2.length!==1){
this.q1.push(this.q2.shift());
}
return this.q2.shift();
}
else{
while(this.q1.length!==1){
this.q2.push(this.q2.shift());
}
return this.q1.shift();
}
}
}