面试题(九)用两个栈实现队列

题目:
用两个栈实现一个队列,需要实现队列的在尾部插入节点和在头部删除节点的功能。队列的声明如下:

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();

    public void push(int node) {

    }

    public int pop() {

    }
}

思路:
通过列举具体的例子来分析两个栈实现的队列的插入和删除元素的过程(将stack1用作插入元素的栈,stack2用作删除元素的栈),可以总结发现:
(1)若stack2不为空,则stack2的栈顶元素即为最先进入队列的元素,将其直接弹出。
(2)若stack2为空,则stack1底部的元素即为最先进入队列的元素,将stack1中的元素全部弹出到stack2中,则位于stack2栈顶的元素即为最先入队的元素,将其直接弹出。

代码(已在牛客网AC):

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.empty()){
            while(!stack1.empty())
                stack2.push(stack1.pop());
        }
        return stack2.pop();

    }
}

扩展题目:
用两个队列实现栈。

思路(举例说明):
将元素a、b、c插入queue1,此时若要删除c,则应该将a、b移动到queue2,再从queue1中删除c。此时queue2中从头至尾元素为a、b。此时若要删除b,则应该将a移至queue1,再从queue2中删除b。此时若要向栈中插入元素,就可以直接入队queue1,因为此时queue1不为空。后续的插入和删除操作重复上述过程。

代码(含单元测试):

import java.util.LinkedList;

public class TwoQueueToStack{

    private LinkedList<Integer> queue1 = new LinkedList<>();
    private LinkedList<Integer> queue2 = new LinkedList<>();

    public static void main(String[] args){
        TwoQueueToStack stack = new TwoQueueToStack();
        stack.test1();
        stack.test2();
        stack.test3();
        stack.test4();
    }

    public void push(int node){
        if(queue1.size() != 0)
            queue1.addLast(node);
        else if(queue2.size() != 0)
            queue2.addLast(node);
        else
            queue1.addLast(node);
    }

    public int pop(){

        if(queue1.size() != 0){
            while(queue1.size() != 1)
                queue2.addLast(queue1.pollFirst());
            return queue1.pollFirst();
        }
        else if(queue2.size() != 0){
            while(queue2.size() != 1)
                queue1.addLast(queue2.pollFirst());
            return queue2.pollFirst();
        }
        else
            return -1;      
    }

    public boolean isEmpty(){
        if(queue1.size() == 0 && queue2.size() == 0)
            return true;
        return false;
    }

    public void test1(){
        push(1);
        push(2);
        push(3);
        push(4);
        push(5);
        while(!isEmpty())
            System.out.printf("%d ", pop());
        System.out.println();
    }

    public void test2(){
        push(1);
        push(2);
        push(3);
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        push(4);
        push(5);
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.println();
    }

    public void test3(){
        push(1);
        push(2);
        push(3);
        System.out.printf("%d ", pop());
        push(4);
        push(5);
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.println();
    }

    public void test4(){
        push(1);
        push(2);
        push(3);
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.printf("%d ", pop());
        System.out.println();
    }
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值