实现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 。

    提示:

    -231 <= val <= 231 - 1
    最多调用 insert、remove 和 getRandom 函数 2 * 105 次
    在调用 getRandom 方法时,数据结构中 至少存在一个 元素。


    1. class RandomizedSet {
    2. Map<Integer, Integer> map;
    3. List<Integer> list;
    4. Random random;
    5. public RandomizedSet() {
    6. map = new HashMap<>();
    7. list = new ArrayList<>();
    8. random = new Random();
    9. }
    10. public boolean insert(int val) {
    11. if (map.containsKey(val)) return false;
    12. map.put(val, list.size());
    13. list.add(val);
    14. return true;
    15. }
    16. public boolean remove(int val) {
    17. if (!map.containsKey(val)) return false;
    18. int idx = map.get(val);
    19. map.put(list.get(list.size() - 1), idx);
    20. list.set(idx, list.get(list.size() - 1));
    21. list.remove(list.size() - 1);
    22. map.remove(val);
    23. return true;
    24. }
    25. public int getRandom() {
    26. return list.get(random.nextInt(list.size()));
    27. }
    28. }
    29. /**
    30. * Your RandomizedSet object will be instantiated and called as such:
    31. * RandomizedSet obj = new RandomizedSet();
    32. * boolean param_1 = obj.insert(val);
    33. * boolean param_2 = obj.remove(val);
    34. * int param_3 = obj.getRandom();
    35. */