LeetCode 232. 用栈实现队列

Categories:

Tags:

题目描述
思路分析
参考代码

题目描述

✅ 232. 用栈实现队列

思路分析

双栈法

参考代码

type MyQueue struct {
	inStack  []int // 输入栈
	outStack []int // 输出栈
}

func Constructor() MyQueue {
	return MyQueue{
		inStack:  []int{},
		outStack: []int{},
	}
}

// 将元素推入队尾
func (q *MyQueue) Push(x int) {
	q.inStack = append(q.inStack, x)
}

// 从队首弹出元素
func (q *MyQueue) Pop() int {
	// 如果 outStack 为空，则将 inStack 元素全部转移
	if len(q.outStack) == 0 {
		for len(q.inStack) > 0 {
			n := len(q.inStack)
			q.outStack = append(q.outStack, q.inStack[n-1])
			q.inStack = q.inStack[:n-1]
		}
	}
	n := len(q.outStack)
	val := q.outStack[n-1]
	q.outStack = q.outStack[:n-1]
	return val
}

// 获取队首元素
func (q *MyQueue) Peek() int {
	if len(q.outStack) == 0 {
		for len(q.inStack) > 0 {
			n := len(q.inStack)
			q.outStack = append(q.outStack, q.inStack[n-1])
			q.inStack = q.inStack[:n-1]
		}
	}
	return q.outStack[len(q.outStack)-1]
}

// 判断队列是否为空
func (q *MyQueue) Empty() bool {
	return len(q.inStack) == 0 && len(q.outStack) == 0
}

➡️ 点击查看 Java 题解

write your code here

本文作者: HGNULB

本文链接: https://hgnulb.github.io/blog/2025/06263683