Java 用两个队列实现一个栈,要求top()的时间复杂度为O(1)

如题:用两个队列实现一个栈,要求top()的时间复杂度为O(1),这就要求top元素不是每次去遍历获取到,而是由一个变量存储。2个队列实现栈的关键点在用队列虽然是先进先出,栈是先进后出,但是2个队列,可以把一个队列的元素都弹出到另一个队列,剩下最后一个元素就是栈顶元素。

栈接口:

/**
 * Create by zxb on 2017/9/3
 */
public interface IStack<T> {

    void push(T newElement);

    void pop();

    T top();

    int size();
}
栈实现类:

/**
 * 用两个队列实现一个栈,要求top()的时间复杂度为O(1)
 * Create by zxb on 2017/9/10
 */
public class StackImpl<T> implements IStack<T> {

    private QueueImpl<T> q1;

    private QueueImpl<T> q2;

    private T topRecord;

    private int size;

    public StackImpl() {
        this.q1 = new QueueImpl<>();
        this.q2 = new QueueImpl<>();
        this.size = 0;
    }

    @Override
    public void push(T newElement) {
        if (q2.size() == 0) {
            q1.push(newElement);
        } else if (q1.size() == 0) {
            q2.push(newElement);
        }
        topRecord = newElement;
        size++;
    }

    @Override
    public void pop() {
        if (size <= 0) {
            return;
        }
        if (q2.size() == 0) {  //从q1导到q2
            if (q1.size() == 1) {  //只有一个元素
                q1.pop();
                topRecord = null;
            }
            while (q1.size() > 1) {
                T front = q1.front();
                //倒2元素为新的栈顶元素
                if (q1.size() == 2) {
                    topRecord = front;
                }
                q2.push(front);
                q1.pop();
            }
            //弹出最后一个元素
            q1.pop();
        } else if (q1.size() == 0) { //从q2导到q1
            if (q2.size() == 1) {  //只有一个元素
                q2.pop();
                topRecord = null;
            }
            while (q2.size() > 1) {
                T front = q2.front();
                //倒2元素为新的栈顶元素
                if (q2.size() == 2) {
                    topRecord = front;
                }
                q1.push(front);
                q2.pop();
            }
            //弹出最后一个元素
            q2.pop();
        }
        size--;
    }

    @Override
    public T top() {
        return topRecord;
    }

    @Override
    public int size() {
        return size;
    }
}
测试类:

/**
 * Create by zxb on 2017/9/10
 */
public class StackTest {

    public static void main(String[] args) {
        IStack stack = new StackImpl();
        stack.push(1);
        stack.push(2);
        stack.push(3);
        stack.push(4);
        stack.pop();
        stack.pop();
        System.out.println("Top:" + stack.top());
    }
}
运行结果:



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值