剑指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!”);
}