题解 | #用两个栈实现队列#

用两个栈实现队列

http://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6

入队直接入

出队判断当前状态:只有第二栈为空时才从第一栈一次性取得所有数据

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

    int pop() {
      if (stack2.empty()) {
        while (!stack1.empty()) {
          stack2.push(stack1.top());
          stack1.pop();
        }
      } 
      
      int res = stack2.top();
      stack2.pop();
      
      return res;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务