原题
Design a data structure that supports all following operations in average O(1) time.
1,insert(val): Inserts an item val to the set if not already present.
2,remove(val): Removes an item val from the set if present.
3,getRandom: Returns a random element from current set of elements. Each element must have the same probability of being returned.
题目翻译
设计一个数据结构,该数据可以在O(1)时间内完成以下操作:
1,insert(val):插入一个元素到集合中,如果元素之前不存在。
2,remove(val):如果val在集合中存在,则删除
3,getRandom:返回当前集合中随机的一个数。每个元素返回的概率相同。
思路解析
由于题目要求删除和插入的时间复杂度均为O(1)。首先想到使用数组保存所有数据。这样的话,插入时间复杂度为O(n)。但是删除数据的时间复杂度为O(n)。随机返回操作时间复杂度为O(n)。如何实现O(1)。我们使用一个map来保存数据-位置关系。然后使用vector来保存元素。
1,insert操作。首先判断val在map中是否存在,时间复杂度为O(1)。如果存在,则返回false。否则,将数据插入到vector中。然后在map中保存数据-位置键值对。时间复杂度为O(1)
2,