O(1) 时间插入、删除和获取随机元素。实现RandomizedSet 类:RandomizedSet() 初始化 RandomizedSet 对象。bool insert(int val) 当元素 val 不存在时,向集合中插入该项,并返回 true ;否则,返回 false 。bool remove(int val) 当元素 val 存在时,从集合中移除该项,并返回 true ;否则,返回 false 。int getRandom() 随机返回现有集合中的一项(测试用例保证调用此方法时集合中至少存在一个元素)。每个元素应该有 相同的概率 被返回。你必须实现类的所有函数,并满足每个函数的 平均 时间复杂度为 O(1) 。力扣380。
两张哈希表。
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]
}
执行结果如下: