《Cracking the Coding Interview》——第3章：栈和队列——题目5

2014-03-18 05:33

题目：用两个栈来实现一个队列。

解法：栈是反的，队列是正的，反了再反就正过来了。所以，请看代码。操作中时间复杂度有O(1)的，有O(n)的，但均摊下来时间符合O(1)。

代码：

 // 3.5 Implement a queue MyQueue using two stacks.
 #include <climits>
 #include <cstdio>
 #include <cstring>
 #include <stack>
 using namespace std;

 template<class T>
 class MyQueue {
 public:
     bool empty() {
         return s1.empty() && s2.empty();
     }

     void push(const T &val) {
         s1.push(val);
     }

     void pop() {
         if (s2.empty()) {
             while (!s1.empty()) {
                 s2.push(s1.top());
                 s1.pop();
             }
         }
         s2.pop();
     }

     T front() {
         if (s2.empty()) {
             while (!s1.empty()) {
                 s2.push(s1.top());
                 s1.pop();
             }
         }

         return s2.top();
     }

     T back() {
         if (s1.empty()) {
             while (!s2.empty()) {
                 s1.push(s2.top());
                 s2.pop();
             }
         }

         return s1.top();
     }
 private:
     stack<T> s1, s2;
 };

 int main()
 {
     MyQueue<int> qq;
     char s[];
     int op;

     while (scanf("%s", s) == ) {
         if (strcmp(s, "end") == ) {
             break;
         } else if (strcmp(s, "push") == ) {
             scanf("%d", &op);
             qq.push(op);
         } else if (strcmp(s, "pop") == ) {
             qq.pop();
         } else if (strcmp(s, "front") == ) {
             printf("%d\n", qq.front());
         } else if (strcmp(s, "back") == ) {
             printf("%d\n", qq.back());
         }
     }

     return ;
 }