【刷算法】用两个栈实现队列

最新推荐文章于 2023-10-25 17:17:37 发布

转载最新推荐文章于 2023-10-25 17:17:37 发布 · 61 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://segmentfault.com/a/1190000015712347

本文介绍了一种使用两个栈来实现队列的方法。通过在两个栈之间转移元素以确保队列的先进先出特性。文章提供了具体的代码实现，并解释了在不同情况下如何正确地进行元素的压入和弹出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。队列中的元素为int类型。

分析

两个栈实现队列，那就是一个进一个出，且题中没说两个栈的长度限制，那么对于两个栈s1和s2来说，有两个注意的点：

s1.push时，若s2不为空且s1为空，为了保证正确的顺序，则将s2的元素全返入s1，然后再s1.push
s2.pop时，若s2为空且s1不为空，为了保证正确的顺序，则将s1的元素全放入s2，然后再s2.pop

代码实现

var s1 = [], s2 = [];
function transferStack(a, b) {
    while(a.length !== 0){
        b.push(a.pop());
    }
}
function push(val) {
    if(s1.length === 0 && s2.length !== 0){
        transferStack(s2, s1);
    }

    s1.push(val);
}

function pop(){
    if(s1.length !== 0 && s2.length === 0){
        transferStack(s1, s2);
    }

    return s2.pop();
}