牛客网刷题—剑指OFFER 005 用两个栈实现队列

一、题目描述

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

二、思路

栈   ： 先进后出    两个栈 s1，s2

队列：先进先出

Push操作 ：  直接压入栈s1

Pop操作   ：  s2不为空直接Pop,否则将s1中的全部压入在 Pop

                1）   s1,s2 都为空，printf " the quene is empty"

                2 )    if     s2不为空，则直接Pop  

                3 )    if   s2为空，push s1 至 s2 ,直到s1为空，Pop

三、代码实现

class Solution
{
public:
    void push(int node) {  
  
    stack1.push(node);

    }

    int pop(){
    int result;
              
    if(stack2.empty()&&stack1.empty())    //都空的话提示队列为空
    {  
        printf("the quene is empty"); 
        return result;                   //因为要有个返回值 ，所以这一步需要
    }
        
    else{
       if (stack2.empty())          //S2为空 s1压入
        {
          while(!stack1.empty())
            {
              stack2.push(stack1.top());
              stack1.pop();
            }         
        }
        result=stack2.top();     //运行到这里说明S2肯定非空，直接弹出即可
        stack2.pop(); 
        return result; 
         }
    }
  
private:
    stack<int> stack1;
    stack<int> stack2;
};