第二题:如何使用栈实现队列(使用两个栈实现队列)

题目要求:

编写一个类,用两个栈实现队列,支持队列的基本操作(add,poll,peek)

 代码实现与分析:

package com.isea.brush;

import java.util.Stack;

/**
 * 编写一个类,用两个栈实现队列,支持队列的基本操作(add,poll,peek)
 * 实现思路:一个stackData栈用来入栈,一个stackHelp用来出栈,每一次入栈,直接将数据存放进入dataStack即可
 * 每一个出栈,判断stackHelp栈是否为空,如果是空,将stackData栈中的内容倾倒到stackHelp栈中,然后再出栈,否者直接出栈
 * 这里发生的倾倒操作必须要stackHelp栈为空。
 */
public class TwoStackQueue {
    private Stack<Integer> stackData;
    private Stack<Integer> stackHelp;

    public TwoStackQueue() {
        stackData = new Stack<Integer>();
        stackHelp = new Stack<Integer>();
    }

    /**
     * 入队操作
     *
     * @param num
     */
    public void add(int num) {
        stackData.push(num);
    }

    /**
     * 出队操作
     *
     * @return
     */
    public int poll() {
        if (stackData.isEmpty() && stackHelp.isEmpty()) {
            throw new IllegalArgumentException("The queue is empty...");
        }
        dump();
        return stackHelp.pop();
    }

    /**
     * 将stackData中的数据倾倒入stackHelp中
     */
    private void dump() {
        if (stackHelp.isEmpty()) {
            while (!stackData.isEmpty()) {
                stackHelp.add(stackData.pop());
            }
        }
    }

    /**
     * 查看队首的元素
     *
     * @return
     */
    public int peek() {
        if (stackHelp.isEmpty() && stackData.isEmpty()){
            throw new IllegalArgumentException("The queue is empty...");
        }
        dump();
        return stackHelp.peek();
    }
    
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值