用两个栈实现队列

1.插入元素进入队列时,将元素push到stack1中。当有元素出队列时,先将所有元素插入到stack2中,然后进行出栈。出栈结束后,再将剩余元素放回stack1中。这个算法不是最优的。接下来介绍一个从剑指offer中看到的一种算法。

2

输入图片说明 

入队列:直接将元素插入stack1中。 出队列:如果stack2为空,就将stack1中的全部元素,压入stack2中。如果stack2还为空,就说明当前队列为空。如果stack2不为空,就弹出stack2栈顶元素。 

 

 

import java.util.Stack;
 
public class Queue {
     
    private Stack<Integer> stack1 = new Stack<Integer>();
    private Stack<Integer> stack2 = new Stack<Integer>();
     
    public void appendTail(int element){
        stack1.push(element);
    }
     
    public int deleteHead(){
        if(stack2.size() <= 0){
            while(stack1.size() > 0){
                int data = stack1.pop().intValue();
                stack2.push(data);
            }
        }
         
        if(stack2.size() == 0){
            System.out.println("队列为空");
            return -1;
        }
         
        int data = stack2.pop().intValue(); 
        return data;
    }
     
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Queue q = new Queue();
        q.appendTail(1);
        q.appendTail(2);
        q.appendTail(3);
        System.out.println(q.deleteHead());
        System.out.println(q.deleteHead());
        System.out.println(q.deleteHead());
        //q.deleteHead();
    }
 
}

  

/**思想:
*入队:直接将元素入栈1;
*出队:判断栈2是否为空,若为空,则将栈1中的元素全部pop,然后push到栈2,栈2出栈;若栈2不为空,则栈2直接出栈。
*/
import java.util.Stack;
public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    
    public void push(int node) {
        stack1.push(node);
    }
    
    public int pop() {
    if(stack2.isEmpty()) {
            while(!stack1.isEmpty()) {
                int temp = stack1.pop();
                stack2.push(temp);
            }  
        } 
        return stack2.pop();
    }
}
 
 
扩展:用两队列1、2实现栈
 
思想:
入栈:直接将元素push到队列1;
出栈:判断队列1中是否只有一个元素,若是,则队列1出队pop;若不是,则将队列1中元素依次出队并入队到队列2,直到队列1中只剩一个元素,然后队列1出队pop,然后将队列2中全部元素出队并入队到队列1;
 
优化:为避免出栈时,最后将队列2中元素频繁出队并入队到队列1。根据出栈后,每次有一个队列为空, 可做以下调整:
入栈:先判断队列1是否为空,若不为空,则直接 将元素push到队列1 ;若队列1为空,则 将元素push到队列2;
出栈:先判断队列1是否为空。
       若队列1 不为空,再判断队列1中是 否只有一个元素,若是,则队列1出队;若不是,则将队列1中元素依次出队并入队到队列2,直到队列1中只剩一个元素,然后队列1出队,此时队列1为空;
        若队列1为空,再判断队列2中是否只有一个元素,若是,则队列2出队;若不是,则将队列2中元素依次出队并入队到队列1,直到队列2中只剩下一个元素,然后队列2出队,此时队列2为空。
 
 
import java.util.Stack;
public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    
    public void push(int node) {
        stack1.push(node);
    }
    
    public int pop() {
    if(stack2.isEmpty()) {
            while(!stack1.isEmpty()) {
                int temp = stack1.pop();
                stack2.push(temp);
            }  
        } 
        return stack2.pop();
    }
}

 

posted @ 2016-08-26 02:41  鹏鹏进阶  阅读(180)  评论(0编辑  收藏  举报