剑指offer第五题

最新推荐文章于 2024-10-20 14:28:59 发布

原创最新推荐文章于 2024-10-20 14:28:59 发布 · 410 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈

java 同时被 2 个专栏收录

15 篇文章

订阅专栏

算法

11 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法。通过这种方式，可以有效地完成队列的Push和Pop操作。当进行Pop操作时，根据辅助栈是否为空采取不同策略，确保元素能够按先进先出的原则出队。

题目

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

思路

本来想的很简单，从stack1往stack2中一倒就把“后进先出”倒成了“先进先出”。
但是这个思路有一个问题，就是stack2里面有元素的时候。本来他应该先出的，反倒在stack2中先进，变成了后出
所以应该把stack2分成两种情况
一种它为空，用原来的思路，另一种它不为空，就直接从stack2里面pop出元素来

import java.util.Stack;

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

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

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

    }

}