【数组】Leetcode 380. O(1) 时间插入、删除和获取随机元素【中等】

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

实现RandomizedSet 类：

• RandomizedSet() 初始化 RandomizedSet 对象
• bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true ；否则，返回 false 。
• bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true ；否则，返回 false 。
• int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有 相同的概率 被返回。

你必须实现类的所有函数，并满足每个函数的 平均 时间复杂度为 O(1) 。

示例：

输入
[“RandomizedSet”, “insert”, “remove”, “insert”, “getRandom”, “remove”, “insert”, “getRandom”]
[[], [1], [2], [2], [], [1], [2], []]
输出
[null, true, false, true, 2, true, false, 2]

解释
RandomizedSet randomizedSet = new RandomizedSet();
randomizedSet.insert(1); // 向集合中插入 1 。返回 true 表示 1 被成功地插入。
randomizedSet.remove(2); // 返回 false ，表示集合中不存在 2 。
randomizedSet.insert(2); // 向集合中插入 2 。返回 true 。集合现在包含 [1,2] 。
randomizedSet.getRandom(); // getRandom 应随机返回 1 或 2 。
randomizedSet.remove(1); // 从集合中移除 1 ，返回 true 。集合现在包含 [2] 。
randomizedSet.insert(2); // 2 已在集合中，所以返回 false 。
randomizedSet.getRandom(); // 由于 2 是集合中唯一的数字，getRandom 总是返回 2 。

解题思路

• 使用哈希表和动态数组来实现 RandomizedSet 类。
• 哈希表用于快速查找元素是否存在，动态数组用于支持随机返回元素。
• 具体而言，哈希表的键存储元素的值，值存储元素在动态数组中的索引。

Java实现

public class RandomizedSet {
    private Map<Integer, Integer> map; // 存储元素值及其在动态数组中的索引
    private List<Integer> list; // 动态数组

    /** 初始化 */
    public RandomizedSet() {
        map = new HashMap<>();
        list = new ArrayList<>();
    }
    
    /** 插入元素.  */
    public boolean insert(int val) {
        if (map.containsKey(val)) {
            return false;
        }
        list.add(val);
        map.put(val, list.size() - 1);
        return true;
    }
    
    /** 删除元素 */
    public boolean remove(int val) {
        if (!map.containsKey(val)) {
            return false;
        }
        int idx = map.get(val);
        int lastElement = list.get(list.size() - 1);
        list.set(idx, lastElement); // 将待删除元素与末尾元素交换位置
        map.put(lastElement, idx); // 更新末尾元素在哈希表中的索引
        list.remove(list.size() - 1); // 删除末尾元素
        map.remove(val); // 删除待删除元素在哈希表中的索引
        return true;
    }
    
    /** 获取随机元素. */
    public int getRandom() {
        Random rand = new Random();
        int idx = rand.nextInt(list.size());
        return list.get(idx);
    }

    public static void main(String[] args) {
        RandomizedSet randomizedSet = new RandomizedSet();
        System.out.println("Test Case 1:");
        System.out.println("Insert 1: " + randomizedSet.insert(1)); // Expected: true
        System.out.println("Insert 2: " + randomizedSet.insert(2)); // Expected: true
        System.out.println("Insert 2: " + randomizedSet.insert(2)); // Expected: false
        System.out.println("Remove 2: " + randomizedSet.remove(2)); // Expected: true
        System.out.println("Random: " + randomizedSet.getRandom()); // Expected: 1 or 2

        System.out.println("\nTest Case 2:");
        System.out.println("Insert 3: " + randomizedSet.insert(3)); // Expected: true
        System.out.println("Insert 4: " + randomizedSet.insert(4)); // Expected: true
        System.out.println("Insert 5: " + randomizedSet.insert(5)); // Expected: true
        System.out.println("Remove 5: " + randomizedSet.remove(5)); // Expected: true
        System.out.println("Random: " + randomizedSet.getRandom()); // Expected: 1, 3, or 4
    }
}

时间空间复杂度

• 时间复杂度： 插入、删除和随机返回操作的平均时间复杂度为 O(1)。
• 空间复杂度： 使用了哈希表和动态数组，空间复杂度为 O(n)，其中 n 是集合中的元素数量。