1. 程式人生 > >經典面試題,兩棧實現佇列以及兩佇列實現棧

經典面試題,兩棧實現佇列以及兩佇列實現棧

經典面試題,兩個棧實現一個佇列,以及兩個佇列實現一個棧。
1、兩個棧實現一個佇列
(1)思路:兩個stack1,stack2,用stack1存放入佇列的資料(入隊操作);stack2負責出佇列的資料,若stack2中有資料就直接出棧,否則把stack1中的資料彈出到stack2中,這樣stack1中底部的數就到了stack2中的頂部,這樣再彈出stack2中的資料即可(出隊操作)。
(2)示例圖
在這裡插入圖片描述
(3)程式碼:

 public class Main {
    	Stack<Integer> stack1=new Stack<Integer>();
    	Stack<Integer> stack2=new Stack<Integer>();
    	/******************
        * 兩棧實現一個佇列,新增操作
        * 只向棧1中新增資料
        * @param data
        */
        public void offerqueue(int data) {
            stack1.push(data);
            return;
        }
        
        /****************
		 * 兩棧實現一個佇列,出佇列操作
		 * 首先判斷棧2是否為空,不為空直接出棧2
		 * 若棧2為空,棧1資料全部出棧到棧2,再出棧棧2
		 * @return
		 */
        public int pollqueue() throws Exception{
			if(stack1.isEmpty()&&stack2.isEmpty()) {
				throw new Exception("佇列為空");
			}
			if(!stack2.isEmpty()) {
				return stack2.pop();
			}
			while(!stack1.isEmpty()) {
				stack2.push(stack1.pop());
			}
			return stack2.pop();
		}
}

2、兩個佇列實現一個棧
(1)思路:哪個佇列不為空就往哪個佇列裡新增資料進行入棧操作,若都為空預設向queue1中加資料(入棧操作);把不為空的佇列中的資料出佇列至另一佇列中,直至該不為空的佇列只剩下最後一個數時即為最後進來的資料,此時出佇列此資料即可(出棧操作)。
(2)示例圖
在這裡插入圖片描述
(3)程式碼:

public class Main {
	Queue<Integer> queue1=new LinkedList<Integer>();
	Queue<Integer> queue2=new LinkedList<Integer>();
	
	/***************
	 * 兩佇列實現棧,新增操作
	 * 向不為空的佇列裡新增資料,預設為佇列1
	 * @param data
	 */
	public void pushStack(int data) {
		//預設放到queue1中
		if(queue1.isEmpty()&&queue2.isEmpty()) {
			queue1.offer(data);
			return;
		}
		else if(!queue1.isEmpty()) {
			queue1.offer(data);
			return;
		}
		else {
			queue2.offer(data);
			return;
		}
	}
	/***************
	 * 兩佇列實現棧,出棧操作
	 * 先把不為空的隊列出佇列到另一佇列中直到只剩下一個元素時,執行該隊出佇列操作
	 * @return
	 */
	public int popStack() throws Exception{
		if(queue1.isEmpty()&&queue2.isEmpty()) {
			throw new Exception("棧為空");
		}
		if(!queue1.isEmpty()) {
			for(int i=0;i<queue1.size()-1;i++) {
				queue2.offer(queue1.poll());
			}
			return queue1.poll();
		}
		else {
			for(int i=0;i<queue2.size()-1;i++) {
				queue1.offer(queue2.poll());
			}
			return queue2.poll();
		}
	}
}