Java:两个栈实现一个队列
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
/*
* stack1和stack2所使用的方法,均为Java提供的Stack类中自带的方法,而不是递归调用,二者只是名字相同
* */
//进栈
public void push(int node) {
//如果进栈次数多,出栈次数少,则stack2中存有元素,故需要先判断stack2中如果不为空,先把stack2中的元素转移到stack1中
while (!stack2.empty()) {
stack1.push(stack2.pop());
}
stack1.push(node);//把元素进stack1中
}
//出栈
public int pop() {
//把stack1中的元素全部转移到stack2中
while (!stack1.empty()) {
stack2.push(stack1.pop());
}
return stack2.pop();//返回stack2出栈的结果
}
}
测试代码
import java.util.Random;
public class demomain {
public static void main(String[] args) {
Solution s = new Solution();
Random r = new Random();
for (int i = 0; i < 5; i++) {
int data = r.nextInt(100);
System.out.println(data + " 进队");
s.push(data); //生成5个随机数,并进栈
}
System.out.println("============================");
System.out.println(s.pop() + " 出队"); //调用一次pop()方法,把stack1中的元素全部转移到stack2中,并输出栈顶元素
while (!s.stack2.empty()) {
System.out.println(s.pop() + " 出队");//遍历stack2中剩余元素并输出
}
}
}
这篇文章参考了下面文章和视频