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

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

題目描述:

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

  • push(x) -- 元素 x 入棧
  • pop() -- 移除棧頂元素
  • top() -- 獲取棧頂元素
  • empty() -- 返回棧是否為空

注意:

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

 

/*解題思路:利用雙佇列實現
建立兩個佇列 q1,q2
入棧:將元素 x 直接放入 q1 佇列中。
出棧:也就是把 q1 的隊尾元素出佇列,由於佇列只能從隊頭出隊,因此先把 q1 中除了隊尾元素的其他值存到 q2 中,再把隊尾元素也就是棧頂出隊,最後將 q2 中的值存到 q1 中。
獲取棧頂元素:也就是獲取 q1 的隊尾元素

*/
class MyStack {
public:
    queue<int> q1, q2;
    int i = 0;
    int temp;
    int res;
    MyStack() {

    }

    
void push(int x) { q1.push(x); } /** Removes the element on top of the stack and returns that element. */ int pop() { int length1 = q1.size(); i = 0; while(length1 > 1 && i < length1 - 1) { temp = q1.front(); q2.push(temp); q1.pop(); i
++; } res = q1.front(); q1.pop(); i = 0; int length2 = q2.size(); while(i < length2) { temp = q2.front(); q1.push(temp); q2.pop(); i++; } return res; } /** Get the top element. */ int top() { return q1.back(); } /** Returns whether the stack is empty. */ bool empty() { return q1.empty(); } }; /** * Your MyStack object will be instantiated and called as such: * MyStack obj = new MyStack(); * obj.push(x); * int param_2 = obj.pop(); * int param_3 = obj.top(); * bool param_4 = obj.empty(); */