2019独角兽企业重金招聘Python工程师标准>>>
队列:先进先出
栈:先进后出;
实现栈:有队列1有队列2。实现原理,始终保持一个队列为空队列。取元素时,将不为空的队列所有元素减一全部放入另外一个队列。将最后一个元素取出即可。
实现队列:原理:始终只从一个栈中取元素。将元素依次放入A中,取时,再将A中的元素全部放入B中。然后取B中的元素即可。
两个队列实现一个栈 + 两个栈实现一个队列 Java
原创 2017年07月16日 17:18:14 536
面试中常出现让你手写两个队列实现一个栈,两个栈实现一个队列的问题,很是头疼!今天就仔细将我分析,思考过的Java代码给大家分享一下:
(一)两个队列实现一个栈:
两个队列添加元素,哪个队列为空,由于在输出元素时,要进行相应元素的移动(除去尾部元素),所以要在对应不为空的队列进行元素的添加;在输出数据时,要进行两个队列的变相操作,不为空的队列要依次向为空的队列中添加元素,直到尾元素输出即可!
[java] view plain copy
/**
*
* @author superYC
* 两个队列实现一个栈
*
*/
public class TwoQueueImplStack {
Queue<Integer> queue1 = new ArrayDeque<Integer>();
Queue<Integer> queue2 = new ArrayDeque<Integer>();
/*
* 向栈中压入数据
*/
public void push(Integer element){
//两个队列都为空时,优先考虑 queue1
if(queue1.isEmpty() && queue2.isEmpty()){
queue1.add(element);
return;
}
//如果queue1为空,queue2有数据,直接放入queue2
if(queue1.isEmpty()){
queue2.add(element);
return;
}
//如果queue2为空,queue1有数据,直接放入queue1中
if(queue2.isEmpty()){
queue1.add(element);
return;
}
}
/*
* 从栈中弹出一个数据
*/
public Integer pop(){
//如果两个栈都为空,则没有元素可以弹出,异常
if(queue1.isEmpty() && queue2.isEmpty()){
try{
throw new Exception("satck is empty!");
}catch(Exception e){
e.printStackTrace();
}
}
//如果queue1中没有元素,queue2中有元素,将其queue2中的元素依次放入queue1中,直到最后一个元素,弹出即可
if(queue1.isEmpty()){
while(queue2.size() > 1){
queue1.add(queue2.poll());
}
return queue2.poll();
}
//如果queue2中没有元素,queue1中有元素,将其queue1中的元素依次放入queue2中,直到最后一个元素,弹出即可
if(queue2.isEmpty()){
while(queue1.size() > 1){
queue2.add(queue1.poll());
}
return queue1.poll();
}
return (Integer)null;
}
public static void main(String[] args) {
TwoQueueImplStack qs = new TwoQueueImplStack();
qs.push(2);
qs.push(4);
qs.push(7);
qs.push(5);
System.out.println(qs.pop());
System.out.println(qs.pop());
qs.push(1);
System.out.println(qs.pop());
}
}
(二)两个栈实现一个队列:
第一个栈只负责添加元素,第二个栈在弹出元素时,首先判断当前栈是否为空,若为空就直接将其第一个栈中的数据全部压入第二个栈中,然后输出栈顶元素,即可实现队列效果;若第二个栈中有数据,添加直接将其数据压入第一个栈中,输出时直接输出第二个栈顶的元素即可!
[java] view plain copy
/**
*
* @author superYC
* 两个栈实现一个队列
*
*/
public class TwoStackImplQueue {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
/*
* 队列的数据压入过程
*/
public void push(Integer element){
stack1.push(element);
}
/*
* 队列的数据弹出过程
*/
public Integer pop(){
if(stack2.size() <= 0){ //第二个栈为空
while(stack1.size() > 0){ //第一个栈不为空
stack2.push(stack1.pop()); //将其第一个栈的数据压入第二个栈中
}
}
if(stack2.isEmpty()){
try{
throw new Exception("queue is empty");
}catch(Exception e){
//e.printStackTrace();
}
}
Integer head = stack2.pop();
return head;
}
public static void main(String[] args) {
TwoStackImplQueue sq = new TwoStackImplQueue();
sq.push(1);
sq.push(3);
sq.push(5);
sq.push(4);
sq.push(2);
System.out.println(sq.pop());
System.out.println(sq.pop());
sq.push(7);
System.out.println(sq.pop());
}
}