用两个栈实现队列&&用两个队列实现一个栈

一.两个栈实现一个队列

思路:所有元素进stack1，然后全部出stack1并进入stack2.实现队列的先进先出即：若stack2非空，我们需要的恰好在栈顶，出栈;若要给队列添加元素，即先进sack1,要出队时，若stack2不为空就出栈，为空时就把stack1全部进栈到stack2。



ps:图片原创于剑指offer，来自网络

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

/*
* Q 57 用两个栈实现队列
*/

public class QueueImplementByTwoStacks {
private Stack<Integer> stack1;
private Stack<Integer> stack2;

QueueImplementByTwoStacks(){
stack1=new Stack<Integer>();
stack2=new Stack<Integer>();
}

public Integer poll(){
Integer re=null;
if(!stack2.empty()){
re=stack2.pop();
}else{
while(!stack1.empty()){//move to stack2 to make stack1 have only one element.Then pop this element.
re=stack1.pop();
stack2.push(re);
}
if(!stack2.empty()){
re=stack2.pop();
}
}
return re;
}
public Integer offer(int o){
stack1.push(o);
return o;
}

public static void main(String[] args) {
QueueImplementByTwoStacks queue=new QueueImplementByTwoStacks();
List<Integer> re=new ArrayList<Integer>();
queue.offer(1);
queue.offer(2);
queue.offer(3);
re.add(queue.poll());
queue.offer(4);
re.add(queue.poll());
queue.offer(5);
re.add(queue.poll());
re.add(queue.poll());
re.add(queue.poll());
System.out.println(re.toString());
}

}

二.两个队列实现一个栈

思路：假设有两个队列Q1和Q2，当二者都为空时，入栈操作可以用入队操作来模拟，可以随便选一个空队列，假设选Q1进行入栈操作，现在假设a,b,c依次入栈了（即依次进入队列Q1），这时如果想模拟出栈操作，则需要将c出栈，因为在栈顶，这时候可以考虑用空队列Q2，将a，b依次从Q1中出队，而后进入队列Q2，将Q1的最后一个元素c出队即可，此时Q1变为了空队列，Q2中有两个元素，队头元素为a，队尾元素为b，接下来如果再执行入栈操作，则需要将元素进入到Q1和Q2中的非空队列，即进入Q2队列，出栈的话，就跟前面的一样，将Q2除最后一个元素外全部出队，并依次进入队列Q1，再将Q2的最后一个元素出队即可。



ps:图片原创于剑指offer，来自网络

import java.util.LinkedList;

/*
* Q 57 用两个队列实现一个栈
*/
public class StackImplementByTwoQueues {

//use 'queue1' and 'queue2' as a queue.That means only use the method 'addLast' and 'removeFirst'.
private LinkedList<Integer> queue1;
private LinkedList<Integer> queue2;

StackImplementByTwoQueues(){
queue1=new LinkedList<Integer>();
queue2=new LinkedList<Integer>();
}

public Integer pop(){
Integer re=null;
if(queue1.size()==0&&queue2.size()==0){
return null;
}
if(queue2.size()==0){
while(queue1.size()>0){
re=queue1.removeFirst();
if(queue1.size()!=0){//do not add the last element of queue1 to queue2
queue2.addLast(re);
}
}
}else if (queue1.size()==0){
while(queue2.size()>0){
re=queue2.removeFirst();
if(queue2.size()!=0){//do not add the last element of queue2 to queue1
queue1.addLast(re);
}
}
}
return re;
}
public Integer push(Integer o){
if(queue1.size()==0&&queue2.size()==0){
queue1.addLast(o);//queue2.addLast(o); is also ok
}
if(queue1.size()!=0){
queue1.addLast(o);
}else if(queue2.size()!=0){
queue2.addLast(o);
}
return o;
}

public static void main(String[] args) {
StackImplementByTwoQueues stack=new StackImplementByTwoQueues();
int tmp=0;
stack.push(1);
stack.push(2);
stack.push(3);
tmp=stack.pop();
System.out.println(tmp);//3
stack.push(4);
tmp=stack.pop();
System.out.println(tmp);//4
tmp=stack.pop();
System.out.println(tmp);//2
stack.push(5);
stack.push(6);
tmp=stack.pop();
System.out.println(tmp);//6
tmp=stack.pop();
System.out.println(tmp);//5
tmp=stack.pop();
System.out.println(tmp);//1
}
}