LeetCode 380. Insert Delete GetRandom O(1)

LeetCode 380

为了能够random的以O（1）取到一个数，基本的想法是要存在一个数组中，这样产生一个随机数之后，我们可以一O（1）读取这个数。同时要能一O（1）插入和删除，那么们需要快速直到这个数的位置，那么我们需要把这个数的位置记录在一个hash表中。在每次插入和删除之后需要调整这个hash表。如果hash表存的是位置的值，如果我们删除中间的数会造成其他的数的位置有变化，但是删除队尾的数就没有这个问题，这里我们可以做一个交换，把要删除的数交换到队尾，然后再进行操作。

class RandomizedSet:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.dic = collections.defaultdict()
        self.vec = []

    def insert(self, val: int) -> bool:
        """
        Inserts a value to the set. Returns true if the set did not already contain the specified element.
        """
        if val in self.dic:
            return False
        self.dic[val] = len(self.vec)
        self.vec.append(val)
        return True

    def remove(self, val: int) -> bool:
        """
        Removes a value from the set. Returns true if the set contained the specified element.
        """
        if val not in self.dic:
            return False
        index = self.dic[val]
        lastindex = len(self.vec) - 1
        if index == lastindex:
            self.vec.pop()
            del self.dic[val]
        else:
            lastv = self.vec[-1]
            self.vec[index] = lastv
            self.vec.pop()
            del self.dic[val]
        return True

    def getRandom(self) -> int:
        """
        Get a random element from the set.
        """
        nl = len(self.vec)
        index = random.randint(0, nl-1)
        return self.vec[index]