2021-11-10：O(1) 时间插入、删除和获取随机元素。实现RandomizedSet 类：RandomizedSet() 初始化 RandomizedSet 对象。bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true ；否则，返回 false 。bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true ；否则，返回 false 。int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有 相同的概率 被返回。你必须实现类的所有函数，并满足每个函数的 平均 时间复杂度为 O(1) 。力扣380。
答案2021-11-10：
两张哈希表。
v→index。
index→v。
index一定是连续的。
删除中间位置的元素，用最后一个元素顶替。这样index就连续了。
代码用golang编写。代码如下：
package main
import (
"fmt"
"math/rand"
"time"
)
func main() {
rand.Seed(time.Now().Unix())
nrs := NewRandomizedSet()
nrs.insert(6)
nrs.insert(3)
nrs.insert(7)
nrs.insert(9)
nrs.insert(8)
fmt.Println(nrs.getRandom())
}
type RandomizedSet struct {
keyIndexMap map[int]int
indexKeyMap map[int]int
size int
}
func NewRandomizedSet() *RandomizedSet {
res := &RandomizedSet{
}
res.keyIndexMap = make(map[int]int)
res.indexKeyMap = make(map[int]int)
res.size = 0
return res
}
func (this *RandomizedSet) insert(val int) bool {
if _, ok := this.keyIndexMap[val]; !ok {
this.keyIndexMap[val] = this.size
this.indexKeyMap[this.size] = val
this.size++
return true
}
return false
}
func (this *RandomizedSet) remove(val int) bool {
if _, ok := this.keyIndexMap[val]; ok {
deleteIndex := this.keyIndexMap[val]
this.size--
lastIndex := this.size
lastKey := this.indexKeyMap[lastIndex]
this.keyIndexMap[lastKey] = deleteIndex
this.indexKeyMap[deleteIndex] = lastKey
delete(this.keyIndexMap, val)
delete(this.indexKeyMap, lastIndex)
return true
}
return false
}
func (this *RandomizedSet) getRandom() int {
if this.size == 0 {
return -1
}
randomIndex := rand.Intn(this.size)
return this.indexKeyMap[randomIndex]
}
执行结果如下：
