【劍指offer - C++/Java】5、用兩個棧實現佇列
阿新 • • 發佈:2018-11-29
牛客網題目連結:用兩個棧實現佇列
題目描述:
用兩個棧來實現一個佇列,完成佇列的Push和Pop操作。 佇列中的元素為int型別。
1、題目分析
首先我們知道資料結構中棧與佇列的特性是:
- 棧:資料先進後出
- 佇列:資料先進先出
假設現在佇列的兩個基本操作為push與pop,兩個輔助的棧為stack1與stack2。那麼可以這實現佇列的性質:
當我們push資料的時候,只把資料往stack1中壓入。因為壓入到stack1中的資料是先進後出,順序是反的,所以當需要pop資料時,再將stack1中的資料全部倒入(彈出)到stack2,這樣的話,資料的順序就變回來了,此時再在stack2中彈出資料返回即可。例如往stack1中壓入1-5,此時在stack1中由上往下是5-1,當想pop時,先將 stack1中的5-1倒入stack2,那麼在stack2中由上往下就是1-5,此時再讓stack2彈出1即可,這就模擬了先進先出規則。如下圖所示:
以上只是很簡單的情況(第一次倒資料)。還有這種情況,1-5已經倒入到stack2中,然後stcak1中沒有資料了,如果現在push,就將資料壓入到stack1中。如果再pop呢?首先需要判斷stack2中還有沒有資料,如果stack2中還有資料,則繼續彈出stack2中的資料。如果stack2中沒有資料了,則將stack1中的所有資料全部倒入stack2中,然後從stack2中彈出資料(自己想一下這個過程)。那麼這就是一個完全模擬佇列的過程。
2、程式碼
Java程式碼:
import java.util.Stack;
public class Solution {
Stack< Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
stack1.push(node); //stack1 用於push資料
}
/* stack2用於pop資料 */
public int pop() {
/* 對於題來說,可以不用判斷異常情況也可以通過 */
if(stack1.isEmpty( ) && stack2.isEmpty()){
throw new RuntimeException("Queue is empty!");
}
/* 如果Stack2為空,則將stack1中的資料全部倒進stack2 */
if(stack2.isEmpty()){
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}
C++程式碼
class Solution
{
public:
void push(int node) {
/* stack1作為插入資料的棧 */
stack1.push(node);
}
int pop() {
/* stack2作為pop資料的棧 */
int tmp=0;
if(stack2.empty()){
while(!stack1.empty()){
tmp=stack1.top();
stack1.pop();
stack2.push(tmp);
}
}
tmp=stack2.top();
stack2.pop();
return tmp;
}
private:
stack<int> stack1;
stack<int> stack2;
};
3、總結
理解棧與佇列的push與pop操作。
學習探討加:
qq:1126137994
微信:liu1126137994