前言:
哈哈,我有來啦打卡啦,不廢話啦,直接進入正題,哼。
題目描述
用兩個棧來實作一個隊列,完成隊列的Push和Pop操作。 隊列中的元素為int類型。
題目解析
解決這道題之前,我們先來了解兩個概念。
棧:是先進後出的。
隊列:是先進先出的。
明天了這兩個概念就可以做這道題了。
當我們向模拟的隊列插入數 a,b,c 時,假設插入的是 stack1,此時的棧情況為:
棧 stack1:{a,b,c}
棧 stack2:{}
當需要彈出一個數,根據隊列的"先進先出"原則,a 先進入,則 a 應該先彈出。但是此時 a 在 stack1 的最下面,将 stack1 中全部元素逐個彈出壓入 stack2,現在可以正确的從 stack2 中彈出 a,此時的棧情況為:
棧 stack1:{}
棧 stack2:{c,b}
繼續彈出一個數,b 比 c 先進入"隊列",b 彈出,注意此時 b 在 stack2 的棧頂,可直接彈出,此時的棧情況為:
棧 stack1:{}
棧 stack2:{c}
此時向模拟隊列插入一個數 d,還是插入 stack1,此時的棧情況為:
棧 stack1:{d}
棧 stack2:{c}
彈出一個數,c 比 d 先進入,c 彈出,注意此時 c 在 stack2 的棧頂,可直接彈出,此時的棧情況為:
代碼樣例:
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);
}
public int pop() {
if (stack2.size() <= 0) {
while (stack1.size() != 0) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}