牛客-NC76-用两个栈实现队列

NC76. 用两个栈实现队列(easy)


方法一:直接模拟法(自己写的)

思路:stack1拿来存值,stack2拿来取值。push方法非常直接,稍微有点曲折的是pop方法,我们需要做判断:(1)stack2还有上一次stack1倒过来的值,而此时pop只需要将stack2中的栈顶元素pop出来就行。(2)当stack2为空,此时需要将stack1新增的值倒过来,再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) {
   
        // stack1 拿来存值
        stack1.push(node);
    }

    public int pop() {
   
        // stack2拿来取值
        // 判断stack2中是否有值
        if (!stack2.isEmpty()) {
   
            return stack2.pop();
        } else {
   
            // 从stack1中导值
            while (!stack1.isEmpty()) {
   
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

时间复杂度: O(N), N是存放到stack1中的元素,因为极端情况下,可能将N个元素一次性push到stack1,再一次性从stack2中pop出来。
空间复杂度: O(1), 未使用额外的空间。
总结:上一个题解才说到栈实现队列,这就遇上了。盘它,详情请看link

全部评论

相关推荐

10-30 10:16
南京大学 Java
永远的鹅孝子:给南大✌️跪了
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务