每日一题 2020.10.31

381. O(1) 时间插入、删除和获取随机元素 - 允许重复

设计一个支持在平均 时间复杂度 O(1) 下， 执行以下操作的数据结构。

注意: 允许出现重复元素。

1. insert(val)：向集合中插入元素 val。
2. remove(val)：当 val 存在时，从集合中移除一个 val。
3. getRandom：从现有集合中随机获取一个元素。每个元素被返回的概率应该与其在集合中的数量呈线性相关。

思路：用一个数组存储出现过的所有数，对于每个val维护其出现的下标，随机值通过随机下标获得

type RandomizedCollection struct {
    idx map[int]map[int]bool
    nums []int

}


/** Initialize your data structure here. */
func Constructor() RandomizedCollection {
    return RandomizedCollection{map[int]map[int]bool{} , []int{} }
}


/** Inserts a value to the collection. Returns true if the collection did not already contain the specified element. */
func (this *RandomizedCollection) Insert(val int) bool {
    _ , ok := this.idx[val]
    if !ok{
        this.idx[val] = map[int]bool{}   
    }
    this.idx[val][len(this.nums)] = true
    this.nums = append(this.nums,val)
    return len(this.idx[val]) == 1
}


/** Removes a value from the collection. Returns true if the collection contained the specified element. */
func (this *RandomizedCollection) Remove(val int) bool {
    indexs , ok := this.idx[val]
    if !ok{
        return false
    }
    index := 0
    for i := range indexs{
        index = i
        break
    }
    length := len(this.nums)
    //交换
    this.nums[index] = this.nums[length-1]
    //删除索引记录
    delete(this.idx[val],index)
    //删除最后一个数时不需要更新
    if index < length - 1{
        delete(this.idx[this.nums[index]],length-1)
        this.idx[this.nums[index]][index] = true 
    }
    if len(this.idx[val]) == 0{
        delete(this.idx,val)
    }
    this.nums = this.nums[:length-1]
    return true
}


/** Get a random element from the collection. */
func (this *RandomizedCollection) GetRandom() int {
    return this.nums[rand.Intn(len(this.nums))]
}


/**
 * Your RandomizedCollection object will be instantiated and called as such:
 * obj := Constructor();
 * param_1 := obj.Insert(val);
 * param_2 := obj.Remove(val);
 * param_3 := obj.GetRandom();
 */