用两个栈实现队列&&用两个队列实现栈

2018-04-11  本文已影响0人  夏臻Rock

题目:

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

思路:

队列是先进先出,栈是先进后出。
用两个栈来实现队列,可以先进A,出栈A后进入B,最终B的出栈顺序为出队列的顺序。

队列的入队push:元素入栈A
队列的出队pop:若B中有元素,则直接B.pop();若B中无元素,则将A中所有元素出栈,并入栈到B后,再进行B.pop()操作。

解答:

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    //队列是先进先出,栈是先进后出。
    //用两个栈来实现队列,可以先进A,出栈A后进入B,最终B的出栈顺序为出队列的顺序。

    public void push(int node) {
        stack1.push(node);
    }

    public int pop(){
        if (stack2.isEmpty()){
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

拓展:

如何使用两个队列实现栈呢?

思路:两个队列保持其中一个为空。
栈的push:将元素push到空的那个队列中,并将非空队列全部入队到这个队列中。(这样,另个队列就空了。)
栈的pop:将非空队列出队即可。

解答:java版本

class twoQueueToStack{
    Queue<Integer> a = new LinkedList<Integer>();
    Queue<Integer> b = new LinkedList<Integer>();
    //保证有一个队列为空,当插入的时候,把新插入的元素放入空队列中,并出队非空队列入队该队列中
    public void push(int node){
        if(a.isEmpty()){
            a.offer(node);
            while (!b.isEmpty()){
                a.offer(b.poll());
            }
        }else if (b.isEmpty()){
            b.offer(node);
            while(!a.isEmpty()){
                b.offer(a.poll());
            }
        }else{ //两个都为空,随便插入一个
            a.offer(node);
        }
    }
    public int pop(){
        if(a.isEmpty()){
           return b.poll();
        }else{
            return a.poll();
        }
    }
}

上一篇 下一篇

猜你喜欢

热点阅读