110 lines
2.6 KiB
Markdown
110 lines
2.6 KiB
Markdown
---
|
||
tags: ["Go", "基础数据结构", "队列"]
|
||
create time: 2026-05-14 18:00
|
||
---
|
||
|
||
# 队列（Queue）
|
||
|
||
## 什么时候用
|
||
|
||
看到 **BFS、层序遍历、滑动窗口** → 用队列。
|
||
FIFO：先进去的先出来。类比排队打饭，先到先得。
|
||
|
||
---
|
||
|
||
## 模板
|
||
|
||
```go
|
||
type Queue struct {
|
||
data []int // 底层用一个 slice 存数据
|
||
}
|
||
|
||
// Push 入队：往末尾追加一个元素
|
||
func (q *Queue) Push(v int) {
|
||
q.data = append(q.data, v)
|
||
}
|
||
|
||
// Pop 出队：取出第一个元素，再把第一个元素切掉
|
||
func (q *Queue) Pop() int {
|
||
v := q.data[0] // 先拿到值
|
||
q.data = q.data[1:] // 再切除第一个位置
|
||
return v
|
||
}
|
||
|
||
// Front 看队首元素但不取走
|
||
func (q *Queue) Front() int {
|
||
return q.data[0]
|
||
}
|
||
|
||
// Empty 判断是否为空
|
||
func (q *Queue) Empty() bool {
|
||
return len(q.data) == 0
|
||
}
|
||
|
||
// Len 当前有多少个元素
|
||
func (q *Queue) Len() int {
|
||
return len(q.data)
|
||
}
|
||
```
|
||
|
||
**核心思路就两件事：** `append` 往尾巴加，`data[1:]` 从头部踢。简单粗暴有效。
|
||
|
||
---
|
||
|
||
## BFS 用法
|
||
|
||
BFS 的精髓就是：**把当前层的节点全部取出，然后把它们的子节点全部放进队列等下一轮处理。**
|
||
|
||
```go
|
||
q := &Queue{}
|
||
q.Push(root) // 先把根节点放进去
|
||
|
||
for !q.Empty() { // 队列不空就一直处理
|
||
node := q.Front() // 看队首
|
||
q.Pop() // 弹出队首
|
||
|
||
if node.Left != nil {
|
||
q.Push(node.Left) // 左子节点入队
|
||
}
|
||
if node.Right != nil {
|
||
q.Push(node.Right) // 右子节点入队
|
||
}
|
||
}
|
||
```
|
||
|
||
如果要**按层**输出（比如 LeetCode 102），在外层循环里记一下当前层有几个节点，内层只处理这么多：
|
||
|
||
```go
|
||
for !q.Empty() {
|
||
size := q.Len() // 当前层有几个节点
|
||
level := []int{} // 存这一层的结果
|
||
|
||
for i := 0; i < size; i++ {
|
||
node, _ := q.Pop()
|
||
level = append(level, node.Val)
|
||
|
||
if node.Left != nil {
|
||
q.Push(node.Left)
|
||
}
|
||
if node.Right != nil {
|
||
q.Push(node.Right)
|
||
}
|
||
}
|
||
result = append(result, level) // 这一层处理完了，存起来
|
||
}
|
||
```
|
||
|
||
> [!tip] 🔑 关键点
|
||
>
|
||
> `size := q.Len()` 这行必须在内层循环前执行。因为内层会不断 Push 新节点进入队列，如果不提前记录当前层的大小，就会无限循环下去。
|
||
|
||
---
|
||
|
||
## 对应题目
|
||
|
||
| 难度 | 题目 | 要点 |
|
||
|------|------|------|
|
||
| ⭐⭐ | [[102.binary-tree-level-order-traversal]] | 层序遍历标准写法 |
|
||
| ⭐⭐ | [[127.word-ladder]] | BFS 求最短路径 |
|
||
| ⭐⭐⭐ | [[239.sliding-window-maximum]] | 单调队列（进阶） |
|