LeetCode232 用棧實現佇列
阿新 • • 發佈:2018-12-03
使用棧實現佇列的下列操作:
- 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 top
,size
, 和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(); */