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
}
}
数据结构与算法之栈与队列
最新推荐文章于 2024-08-26 16:40:46 发布