use 2 stacks to simulate a queue

最新推荐文章于 2021-03-05 14:06:18 发布

转载最新推荐文章于 2021-03-05 14:06:18 发布 · 115 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/miaoz/p/4440544.html

本文介绍了一种使用栈实现队列的方法。通过两个栈s1和s2的配合使用，实现了队列的基本操作：enqueue（入队）、dequeue（出队）、peek（查看队首元素）及size（获取队列长度）。当执行队列操作时，元素会在两个栈之间转移，以保证队列的先进先出特性。

class Stack{
private:
    int cur = 0;
    int elem[2000001];
public:
    void push(int n);
    int pop();
    int peek();
    int size();
};

void Stack::push(int n){
    elem[cur] = n;
    cur++;
}

int Stack::pop(){
    cur--;
    return elem[cur + 1];    
}

int Stack::peek(){
    return elem[cur - 1];
}

int Stack::size(){
    return cur;
}

class Queue{
private:
    Stack s1, s2;
public:
    void enqueue(int n);
    int dequeue();
    int peek();
    int size();    
};

void Queue::enqueue(int n){
    if(s1.size() == 0){
        while(s2.size()){
            s1.push(s2.pop());
        }
    }

    s1.push(n);
}

int Queue::dequeue(){
    if(s2.size() == 0){
        while(s1.size()){
            s2.push(s1.pop());
        }
    }
    return s2.pop();
}

int Queue::peek(){
    if(s2.size() == 0){
        while(s1.size()){
            s2.push(s1.pop());
        }
    }

    return s2.peek();
}

int Queue::size(){
    return s1.size()? s1.size() : s2.size();
}