1. 程式人生 > >LeetCode232 棧·用棧實現佇列(C++)

LeetCode232 棧·用棧實現佇列(C++)

題目描述:

使用棧實現佇列的下列操作:

  • push(x) -- 將一個元素放入佇列的尾部。
  • pop() -- 從佇列首部移除元素。
  • peek() -- 返回佇列首部的元素。
  • empty() -- 返回佇列是否為空。

示例:

MyQueue queue = new MyQueue();

queue.push(1);
queue.push(2);  
queue.peek();  // 返回 1
queue.pop();   // 返回 1
queue.empty(); // 返回 false

說明:

  • 你只能使用標準的棧操作 -- 也就是隻有 push to top
    peek/pop from topsize, 和 is empty 操作是合法的。
  • 你所使用的語言也許不支援棧。你可以使用 list 或者 deque(雙端佇列)來模擬一個棧,只要是標準的棧操作即可。
  • 假設所有操作都是有效的 (例如,一個空的佇列不會呼叫 pop 或者 peek 操作)。

 

/*解題思路:
由於佇列是先進先出的,而棧是先進後出的,所以要用2個棧來實現佇列的入隊出隊功能,佇列的入隊功能與棧的一樣,出隊時,先將第一個棧中的元素全部彈出,並倒入到第二個棧中,將第二個棧中棧頂元素彈出,並將stack2中剩下的元素倒回到stack1中,即實現一次出隊。

*/ class MyQueue { public: /** Initialize your data structure here. */ stack<int> s1, s2; MyQueue() { } /** Push element x to the back of queue. */ void push(int x) { while(!s2.empty()) { s1.push(s2.top()); s2.pop(); } s2.push(x);
while(!s1.empty()) { s2.push(s1.top()); s1.pop(); } } /** Removes the element from in front of queue and returns that element. */ int pop() { int a = s2.top(); s2.pop(); return a; } /** Get the front element. */ int peek() { return s2.top(); } /** Returns whether the queue is empty. */ bool empty() { return s2.empty(); } }; /** * Your MyQueue object will be instantiated and called as such: * MyQueue obj = new MyQueue(); * obj.push(x); * int param_2 = obj.pop(); * int param_3 = obj.peek(); * bool param_4 = obj.empty(); */