劍指offer第5題:
用兩個棧來實現一個隊列,完成隊列的Push和Pop操作。 隊列中的元素爲int類型。
理解:
完成本題首先應當理解堆棧和隊列的關鍵性質。堆棧是先入後出,FILO;而隊列是先入先出,FIFO。
當有元素需要進入隊列就將其壓入Stack1,當有元素需要出隊列,就將Stack1中的元素全部彈出到Stack2中,通過Stack2來負責出隊列。Stack1負責入,Stack2負責出。需要注意的一點是要將Stack2中的全部pop掉之後才需要把Stack1中的元素再彈出壓入一批。
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) { //直接進入堆棧1即可
stack1.push(node);
}
public int pop() throws Exception{
if(stack1.isEmpty() && stack2.isEmpty()){ //如果兩個棧都爲空,則不能執行pop操作了
throw new Exception("No numbers!");
}
if(stack2.isEmpty()){
while(!stack1.isEmpty()){ //當堆棧2空了之後需要將堆棧1的內容彈出壓過來
stack2.push(stack1.pop());
}
}
return stack2.pop(); //注意return
}
}
知識點:
- 堆棧Stack,需要import;
- 堆棧方法isEmpty(),判斷是否爲空,爲空時爲1;
- stack.push(xx); stack.pop();
- public int xxx() throws Exception{
if(xxx){
throw new Exception(“XXX!”);
}