两个栈模拟一个队列(5)

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

比较容易想到的pop实现是pop_1,优化后用pop_2实现,其时间复杂度更低。
pop_1思路:每次都将stack1中的元素都压倒stack2中,然后弹出stack2的栈顶元素,再将stack2中的元素压到stack1中。

pop_2思路:当stack2中有元素的时候直接返回stack2的栈顶元素,当stack2中没有元素的时候就将stack1中的元素压倒stack2中,然后返回stack2的栈顶元素。把握住stack2的栈顶就是队列头部。

显然pop_2的实现方式要比pop_1要好,因为只要stack2中有元素就不用将stack1中的元素倒过来。


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_1() {
        if(stack1.isEmpty()){
            //这里return null;不能通过说是因为null不能转化成int类型,只能抛异常。
            throw new RuntimeException("栈为空");
        }
        while(!stack1.isEmpty()){
            stack2.push(stack1.pop());
        }
        int node=stack2.pop();
        while(!stack2.isEmpty()){
           stack1.push(stack2.pop());
        }
        return node;
        
    }
    
    public int pop_2(){
        if(stack1.isEmpty() && stack2.isEmpty()){
            //这里return null;不能通过说是因为null不能转化成int类型,只能抛异常。
            throw new RuntimeException("栈为空");
        }
        
        if(stack2.isEmpty()){
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
        }
        
        return stack2.pop();
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值