常数时间插入、删除和获取随机元素
O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。
O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。
一次更新后,所有索引和顺序结构是否仍然指向同一对象?
数组存值、map(val→index);删时用末元素填坑再 pop。
先说结论:这道题到底解决什么
怎样从“O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。”推导出 设计 · 数组+哈希,并证明每次状态变化都不会漏掉答案?
中心结论:数组存值、map(val→index);删时用末元素填坑再 pop。
- 1.暴力方案在哪里重复计算,为什么仍然是正确基线?
- 2.一次更新后,所有索引和顺序结构是否仍然指向同一对象?
- 3.不变量“map 与数组内容双向一致。”为什么能保证算法安全前进?
完整题目与题意拆解
O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。
在本站主例中,O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。
算法最终需要得到或观察:remove(2) 后 getRandom 仍在 {1,3} 中等概率。
- • 输入:O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。
- • 机器需要维护:nums 数组、val→idx map、长度。
- • 最终可观察结果:remove(2) 后 getRandom 仍在 {1,3} 中等概率。
先看清算法到底要维护什么
先建立输入、目标、输出和第一批状态,不急着进入模板。
insert/delete/getRandom 均 O(1)
第一层方案:暴力做法
只用一种结构往往让某个操作退化为线性;组合结构让查找、更新和顺序维护共享同一份状态。
暴力方案的价值是确认题意并提供正确性基线。它通常会覆盖所有候选, 但没有保存已经确认的信息,因此同一状态会被重新计算。
重复工作究竟发生在哪里
把重复读取或重复搜索的区域明确标出,再决定优化必须保存什么。
insert/delete/getRandom 均 O(1)
整体地图:先做什么,再做什么
- 1建模把输入翻译成“数据结构控制台”,明确答案需要观察什么。
- 2状态只维护 nums 数组、val→idx map、长度。
- 3转移每一步按照 数组存值、map(val→index);删时用末元素填坑再 pop。
- 4收尾读取 remove(2) 后 getRandom 仍在 {1,3} 中等概率。,并复核边界与复杂度。
数据结构控制台:核心概念
为每个操作写出复杂度目标,再反推需要组合哪些数据结构。
这道题不是为了记住一个组件名称,而是为了让状态具有可解释的语义:nums 数组、val→idx map、长度。
- • map 与数组内容双向一致。
建立“数据结构控制台”心智模型
用主例建立核心状态,先预测下一步,再公开正确分支和理由。
insert/delete/getRandom 均 O(1)
核心机制:状态如何一步步变化
数组存值、map(val→index);删时用末元素填坑再 pop。
执行过程中持续维护:nums 数组、val→idx map、长度。
正确性依赖以下不变量:map 与数组内容双向一致。
面试时可以压缩为:数组+哈希:insert push;remove 用末尾覆盖并 pop;random 按 index 取,O(1) 全操作。
落到当前题,执行机制可以压缩为:数组存值、map(val→index);删时用末元素填坑再 pop。每一次更新都必须保持核心不变量,而不是只让样例碰巧得到正确答案。
一次状态转移为什么成立
集中观察一次选择、计算和状态写回,让变量变化与原因同时出现。
arr 尾部追加 1
正确性证明:为什么不会漏答案
初始化:算法开始时,全部合法候选仍在状态表示范围内;map 与数组内容双向一致。
保持:执行“数组存值、map(val→index);删时用末元素填坑再 pop。”时,只删除已经能证明不可能的候选,并把新信息写回 nums 数组、val→idx map、长度。
终止:没有待处理状态或达到命中条件时,当前可观察结果就是“remove(2) 后 getRandom 仍在 {1,3} 中等概率。”。
完整执行过程
- 1题目与输入O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。 因为:数组存值、map(val→index);删时用末元素填坑再 pop。
- 2数组存值 + map 存下标RandomizedSet 核心。 因为:delete 用最后一个补洞。
- 3insert 2 成功O(1) 插入。 因为:map 记录下标。
- 4预测下一步先不要看下一帧——根据当前不变量,预测算法接下来会怎么动。 因为:主动预测会暴露你对不变量的真实理解,比被动看动画有效得多。
- 5remove 1:用末尾 2 填洞swap-pop 删除。 因为:保持 arr 紧凑。
- 6insert 2 成功O(1) 插入。 因为:map 记录下标。
- 7RandomizedSet 操作完成结构仍 O(1)。 因为:数组+哈希经典设计。
- 8收尾与复杂度remove(2) 后 getRandom 仍在 {1,3} 中等概率。 因为:时间 O(1) 各操作 · 空间 O(n)。数组存值、map(val→index);删时用末元素填坑再 pop。
从输入完整走到可观察结果
从主例第一帧运行到答案,时间轴始终显示当前状态、因果解释和下一步。
insert/delete/getRandom 均 O(1)
把动画和 Go 代码逐行对应
代码窗口不会按容易漂移的固定数字行号硬绑动画,而是根据当前语义阶段, 在完整 Go 实现中定位选择、计算、写回或返回分支。当前变量与高亮行一起变化。
让每个动作都落到 Go 分支
重新执行关键分支,只显示当前代码附近窗口,并说明这行为什么在此刻运行。
insert/delete/getRandom 均 O(1)
完整 Go 提交代码与最小测试
import "math/rand"
type RandomizedSet struct {
values []int
index map[int]int
}
func Constructor() RandomizedSet {
return RandomizedSet{index: map[int]int{}}
}
func (set *RandomizedSet) Insert(val int) bool {
if _, ok := set.index[val]; ok { return false }
set.index[val] = len(set.values)
set.values = append(set.values, val)
return true
}
func (set *RandomizedSet) Remove(val int) bool {
position, ok := set.index[val]
if !ok { return false }
last := set.values[len(set.values)-1]
set.values[position] = last
set.index[last] = position
set.values = set.values[:len(set.values)-1]
delete(set.index, val)
return true
}
func (set *RandomizedSet) GetRandom() int {
return set.values[rand.Intn(len(set.values))]
}func main() {
// 1. 主例
// 输入:mode="insert-delete-random", ops=["insert 1","insert 2","remove 1","insert 2","getRandom"]
// 期望:remove(2) 后 getRandom 仍在 {1,3} 中等概率。
//
// 2. 失败 / 未命中
// 检查:删除中间元素时忘记把末尾换到被删位置。
//
// 3. 边界
// 空输入、单元素、最小合法规模,以及答案恰好落在边界的情况。
//
// 4. 迁移
// LC146 LRU Cache;LC381 O(1) Random with Dup
}正确性与复杂度
执行过程中只保留仍可能影响答案的状态。数组存值、map(val→index);删时用末元素填坑再 pop。
额外状态主要用于维护:nums 数组、val→idx map、长度。
- • map 与数组内容双向一致。
最容易写错的地方
删除中间元素时忘记把末尾换到被删位置。
必须额外检查空输入、单元素、未命中或不可达情况,以及恰好落在边界的输入。
最后复盘:带走逻辑链
- 1题意O(1) 插入/删除/随机:getRandom 等概率返回 {1,2,3} 中任一。
- 2重复只用一种结构往往让某个操作退化为线性;组合结构让查找、更新和顺序维护共享同一份状态。
- 3优化数组存值、map(val→index);删时用末元素填坑再 pop。
- 4证明map 与数组内容双向一致。
- 5复杂度时间 O(1) 各操作,空间 O(n)
- • LC146 LRU Cache
- • LC381 O(1) Random with Dup