数据结构与算法之栈与队列

package StackAndQueueSummary;

import java.util.LinkedList;
import java.util.Queue;

/**
 * 4.用两个队列实现栈
 * 基本思路：使用queue1和queue2两个队列轮流存储数据。
 * 这个轮流发生再pop和peek操作的时候，
 * 假设此时我们把所有的数据存储在queue1中（queue2为空），我们把queue1中的n-1个元素放到queue2中，
 * queue1中的最后一个剩余元素就是我们想要pop的元素，此时queue2存在n-1个元素（queue1变为空）。
 * 至于peek操作，则是先转移n-1个数据，在转第n个数据的时候，将其记下并返回该值。
 * 
 * 对于push操作，只需要判断当前两个队列哪一个不为空，将其添加到不为空的队列中即可。
 * 
 * 
 *
 */
public class NewStackByQueue<E> {
	private Queue<E> queue1;
	private Queue<E> queue2;
	public NewStackByQueue(){
		queue1 = new LinkedList<>();
		queue2 = new LinkedList<>();
	}
	public void push(E item){
		if(queue1.isEmpty()){
			//向不为空的队列添加元素
			queue2.add(item);
		}else{
			queue1.add(item);
		}
	}
	
	public E pop() throws Exception{
		if(queue1.isEmpty() && queue2.isEmpty()){
			throw new Exception("The stack is empty");
		}
		if( !queue1.isEmpty()){
			//queue1不为空
			int len1 = queue1.size();
			//先从queue1中移除n-1个元素到queue2中
			for(int i = 1;i<len1;i++){
				queue2.add(queue1.remove());
			}
			//将queue1中的最后一个剩余元素移除，并返回该值
			return queue1.remove();
		}else{
			//queue2不为空
			int len2 = queue2.size();
			//先从queue2中移除n-1个元素到queue1中
			for(int i = 1;i<len2;i++){
				queue1.add(queue2.remove());
			}
			//将queue2中的最后一个剩余元素移除，并返回该值
			return queue2.remove();
			
		}
	}
	
	public E peek() throws Exception{
		if(queue1.isEmpty() && queue2.isEmpty()){
			throw new Exception("The stack is empty");
		}
		if( !queue1.isEmpty()){
			//queue1不为空
			int len1 = queue1.size();
			//先从queue1中移除n-1个元素到queue2中
			for(int i = 1;i<len1;i++){
				queue2.add(queue1.remove());
			}
			//将queue1中的最后一个剩余元素移到queue2中，并返回该值
			E res = queue1.remove();
			queue2.add(res);
			return res;
		}else{
			//queue2不为空
			int len2 = queue2.size();
			//先从queue2中移除n-1个元素到queue1中
			for(int i = 1;i<len2;i++){
				queue1.add(queue2.remove());
			}
			//将queue2中的最后一个剩余元素移到queue1中，并返回该值
			E res =  queue2.remove();
			queue1.add(res);
			return res;
			
		}
	}
	
	public static void main(String[] args) throws Exception {
		NewStackByQueue<Integer> newStackByQueue = new NewStackByQueue<>();
		newStackByQueue.push(1);
		newStackByQueue.push(2);
		newStackByQueue.push(3);
		newStackByQueue.push(4);
		
		System.out.println(newStackByQueue.pop());//4
		System.out.println(newStackByQueue.pop());//3
		newStackByQueue.push(5);
		newStackByQueue.push(6);
		System.out.println(newStackByQueue.peek());//6
		System.out.println(newStackByQueue.peek());//6
		System.out.println(newStackByQueue.pop());//6
		System.out.println(newStackByQueue.pop());//5
		
		
		
		
	}
	
}