用两个栈实现队列-javascript

89 阅读1分钟

描述

用两个栈来实现一个队列,使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法,即保证pop操作时队列内已有元素。

数据范围: n≤1000 要求:存储n个元素的空间复杂度为 O(n),插入与删除的时间复杂度都是 O(1)

示例 输入: [“PSH1”,“PSH2”,“POP”,“POP”] 返回: 1,2

let stack1=[];
let stack2=[];
function push(node){
    stack1.push(node);
}
function pop(){
     if (stack2.length === 0) {
        while(stack1.length){
            stack2.push(stack1.pop());
        }
     }
    return stack2.pop();
}

思路

  • 当插入时,直接插入 stack1
  • 当弹出时,当 stack2 不为空,弹出 stack2 栈顶元素,如果 stack2 为空,将 stack1 中的全部数逐个出栈入栈 stack2,再弹出 stack2 栈顶元素