《劍指offer》-用兩個棧來實現一個隊列

import java.util.Stack;
/*
 *用兩個棧來實現一個隊列,完成隊列的Push和Pop操作。 隊列中的元素爲int類型。
 *思路:兩次先進後出變成先進先出(畫圖)
 */
public class UseStackImplementQueue {
	Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
    	stack1.push(node);
    }
    
    public int pop() {
    	if(stack1.isEmpty() || !(stack2.isEmpty())) {
    		return -1;
    	}
    	
    	while(!stack1.isEmpty()) {
    		stack2.push(stack1.pop());
    	}
    	
    	int value = stack2.pop();
    	while(!stack2.isEmpty()) {	//恢復現場:始終保持棧1裏存元素,棧2裏不存元素。
    		stack1.push(stack2.pop());
    	}
        
    	return value;
    }
    
    public static void main(String[] args) {
		UseStackImplementQueue t = new UseStackImplementQueue();
		t.push(1);
		t.push(2);
		t.push(3);
		System.out.println(t.pop());
		t.push(4);
		System.out.println(t.pop());
	}
}

發佈了83 篇原創文章 · 獲贊 9 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章