题目: 个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
比较容易想到的pop实现是pop_1,优化后用pop_2实现,其时间复杂度更低。
pop_1思路:每次都将stack1中的元素都压倒stack2中,然后弹出stack2的栈顶元素,再将stack2中的元素压到stack1中。
pop_2思路:当stack2中有元素的时候直接返回stack2的栈顶元素,当stack2中没有元素的时候就将stack1中的元素压倒stack2中,然后返回stack2的栈顶元素。把握住stack2的栈顶就是队列头部。
显然pop_2的实现方式要比pop_1要好,因为只要stack2中有元素就不用将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) {
stack1.push(node);
}
public int pop_1() {
if(stack1.isEmpty()){
//这里return null;不能通过说是因为null不能转化成int类型,只能抛异常。
throw new RuntimeException("栈为空");
}
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
int node=stack2.pop();
while(!stack2.isEmpty()){
stack1.push(stack2.pop());
}
return node;
}
public int pop_2(){
if(stack1.isEmpty() && stack2.isEmpty()){
//这里return null;不能通过说是因为null不能转化成int类型,只能抛异常。
throw new RuntimeException("栈为空");
}
if(stack2.isEmpty()){
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}