Problem: 380. O(1) 时间插入、删除和获取随机元素

思路

👩‍🏫 三叶题解

复杂度

时间复杂度: $O(1)$

空间复杂度: $O(n)$

Code

class RandomizedSet {static int[] nums = new int[200_010];//存元素的值Random random = new Random();Map<Integer, Integer> map = new HashMap<>();//key：元素的值，val：元素在nums数组中的下标int idx = -1;public boolean insert(int val) {if(map.containsKey(val))return false;nums[++idx] = val;map.put(val,idx);return true;}public boolean remove(int val) {if(!map.containsKey(val))return false;int cur = map.remove(val);//删除当前元素在 map 中的键值对，并返回 值（待删除元素在数组中的下标）if(cur != idx)//如果最后一个元素就是要删除的元素，直接跳过map.put(nums[idx],cur);//更新最后一个元素在数组中的下标nums[cur] = nums[idx--];//最后一个元素与当前元素交换，然后删除最后一个元素return true;}public int getRandom() {return nums[random.nextInt(idx+1)];}
}/*** Your RandomizedSet object will be instantiated and called as such:* RandomizedSet obj = new RandomizedSet();* boolean param_1 = obj.insert(val);* boolean param_2 = obj.remove(val);* int param_3 = obj.getRandom();*/