用2个队列实现栈操作

简介:

一、文章来由

一道面试题,别说以前还真没好好想过,在未参考其他资料情况下自己想了两种,实现如下,如果另有高招,之后再补。

二、2种实现方式

分析:栈操作其实只有 push 和 pop 对栈内元素进行改变,我于是想从其一下手即可。

改变 push 操作:

比如输入,1 2 3 4 5,栈中结构应该是1在底,5在顶,但是队中元素是原顺序,如果要让push一个元素,让队列中的顺序反过来,就可以每推入一个元素,就用另一个队列中转把次序调整过来。

/* 
* 
* function: 用两个栈实现队列
* 
* Date:2016-03-15
* 
*    Author: Bill Wang
*/

#include <iostream>
#include <queue>

using namespace std;

template <class T>
class whStack
{
public:
    whStack(){}
    ~whStack(){}

    void push(T t);
    T pop();

private:
    std::queue<T> QueueA;
    std::queue<T> QueueB;
};

template<class T> void whStack<T>::push(T t) {

    //将另一个队列作为中转站

    //step1、将A队列所有元素推入B
    while ( !QueueA.empty() ) {
        T tmp = QueueA.front();
        QueueA.pop();
        QueueB.push(tmp);
    }

    //step2、推入新节点
    QueueA.push(t);

    //step3、将B队列回归A
    while ( !QueueB.empty() ) {
        T tmp = QueueB.front();
        QueueB.pop();
        QueueA.push(tmp);
    }

}

//pop时直接弹出并返回T
template<class T> T whStack<T>::pop() {

    T res = QueueA.front();
    QueueA.pop();
    return res;
}

int main() 
{
    whStack<int > s;

    s.push(1);
    s.push(2);
    s.push(3);
    s.push(4);
    s.push(5);

    cout<<s.pop()<<endl;
    cout<<s.pop()<<endl;
    cout<<s.pop()<<endl;

    return 0; 
}

这个输出:
5
4
3

但是这样相当于把另一个队列纯做中转,时间复杂度较高。

改变 pop 操作:

/* 
* 
* function: 用两个栈实现队列2
* 
* Date:2016-03-15
* 
*    Author: Bill Wang
*/

#include <iostream>
#include <queue>

using namespace std;

template <class T>
class whStack
{
public:
    whStack(){}
    ~whStack(){}

    void push(T t);
    T pop();

private:
    std::queue<T> QueueA;
    std::queue<T> QueueB;
};

template<class T> void whStack<T>::push(T t) {

    //判断哪一个队列非空
    if( !QueueA.empty() )
        QueueA.push(t);
    else
        QueueB.push(t);
}

//pop时直接弹出并返回T
template<class T> T whStack<T>::pop() {


    std::queue<T> &qa = QueueA.empty()? QueueB:QueueA;
    std::queue<T> &qb = QueueA.empty()? QueueA:QueueB;

    //将队尾最后一个元素留住,将其他元素入另一个队列
    T tmp;
    while (1) {

        tmp = qa.front();
        qa.pop();

        if( qa.empty() )
            break;

        qb.push(tmp);
    }

    return tmp;
}

int main() 
{
    whStack<int > s;

    s.push(1);
    s.push(2);
    s.push(3);
    s.push(4);
    s.push(5);

    cout<<s.pop()<<endl;
    cout<<s.pop()<<endl;
    cout<<s.pop()<<endl;

    return 0; 
}

同样输出:
5
4
3

三、复杂度分析

从源码可以看出,如果队列元素个数为n,他们的如果执行一对push和pop,都可以在线性时间 O(n),得出结果。

但是明显,第一种方法是将元素移进移出两次,一个队列始终为空,严格来说复杂度是 O(2n);第二种方法是两个队列轮流使用,省去一轮遍历,所以相较之下应该是 O(n),所以第二种更快一些。

相关文章
|
2天前
|
存储 C语言
数据结构基础详解(C语言): 栈与队列的详解附完整代码
栈是一种仅允许在一端进行插入和删除操作的线性表,常用于解决括号匹配、函数调用等问题。栈分为顺序栈和链栈,顺序栈使用数组存储,链栈基于单链表实现。栈的主要操作包括初始化、销毁、入栈、出栈等。栈的应用广泛,如表达式求值、递归等场景。栈的顺序存储结构由数组和栈顶指针构成,链栈则基于单链表的头插法实现。
|
4天前
|
Java
【数据结构】栈和队列的深度探索,从实现到应用详解
本文介绍了栈和队列这两种数据结构。栈是一种后进先出(LIFO)的数据结构,元素只能从栈顶进行插入和删除。栈的基本操作包括压栈、出栈、获取栈顶元素、判断是否为空及获取栈的大小。栈可以通过数组或链表实现,并可用于将递归转化为循环。队列则是一种先进先出(FIFO)的数据结构,元素只能从队尾插入,从队首移除。队列的基本操作包括入队、出队、获取队首元素、判断是否为空及获取队列大小。队列可通过双向链表或数组实现。此外,双端队列(Deque)支持两端插入和删除元素,提供了更丰富的操作。
10 0
【数据结构】栈和队列的深度探索,从实现到应用详解
|
8天前
|
Linux C++ Windows
栈对象返回的问题 RVO / NRVO
具名返回值优化((Name)Return Value Optimization,(N)RVO)是一种优化机制,在函数返回对象时,通过减少临时对象的构造、复制构造及析构调用次数来降低开销。在C++中,通过直接在返回位置构造对象并利用隐藏参数传递地址,可避免不必要的复制操作。然而,Windows和Linux上的RVO与NRVO实现有所不同,且接收栈对象的方式也会影响优化效果。
|
23天前
|
存储 安全 编译器
缓冲区溢出之栈溢出(Stack Overflow
【8月更文挑战第18天】
46 3
|
24天前
|
测试技术
【初阶数据结构篇】栈的实现(附源码)
在每一个方法的第一排都使用assert宏来判断ps是否为空(避免使用时传入空指针,后续解引用都会报错)。
|
10天前
crash —— 获取内核地址布局、页大小、以及栈布局
crash —— 获取内核地址布局、页大小、以及栈布局
|
10天前
|
存储 程序员 C语言
堆和栈之间有什么区别
【9月更文挑战第1天】堆和栈之间有什么区别
77 0
|
19天前
|
机器学习/深度学习 消息中间件 缓存
栈与队列的实现
栈与队列的实现
35 0
|
24天前
|
测试技术
【初阶数据结构篇】队列的实现(赋源码)
首先队列和栈一样,不能进行遍历和随机访问,必须将队头出数据才能访问下一个,这样遍历求个数是不规范的。
|
29天前
|
算法 C语言 C++
【practise】栈的压入和弹出序列
【practise】栈的压入和弹出序列