面试算法30:插入、删除和随机访问都是O(1)的容器

题目

设计一个数据结构,使如下3个操作的时间复杂度都是O(1)。

  • insert(value):如果数据集中不包含一个数值,则把它添加到数据集中。
  • remove(value):如果数据集中包含一个数值,则把它删除。
  • getRandom():随机返回数据集中的一个数值,要求数据集中每个数字被返回的概率都相同。

分析

由于题目要求插入和删除(包括判断数据集中是否包含一个数值)的时间复杂度都是O(1),能够同时满足这些时间效率要求的只有哈希表,因此这个数据结构要用到哈希表。但是如果只用哈希表,则不能等概率地返回其中的每个数值。

如果数值是保存在数组中的,那么很容易实现等概率返回数组中的每个数值。假设数组的长度是n,那么等概率随机生成从0到n-1的一个数字。如果生成的随机数是i,则返回数组中下标为i的数值。由此可以发现,需要结合哈希表和数组的特性来设计这个数据容器。

由于数值保存在数组中,因此需要知道每个数值在数组中的位置,否则在删除的时候就必须顺序扫描整个数组才能找到待删除的数值,那就需要O(n)的时间。通常把每个数值在数组中的位置信息保存到一个HashMap中,HashMap的键是数值,而对应的值为它在数组中的位置。

java 复制代码
public class Test {
    public static void main(String[] args) {
        RandomizedSet randomizedSet = new RandomizedSet();
        randomizedSet.insert(1);
        randomizedSet.insert(2);
        randomizedSet.insert(3);
        randomizedSet.insert(4);
        for (int i = 0; i < randomizedSet.nums.size(); i++) {
            System.out.println(randomizedSet.nums.get(i));
        }
        System.out.println("-----------------------");
        randomizedSet.remove(2);
        for (int i = 0; i < randomizedSet.nums.size(); i++) {
            System.out.println(randomizedSet.nums.get(i));
        }
        System.out.println("-----------------------");
        System.out.println(randomizedSet.getRandom());
    }

    static class RandomizedSet {
        HashMap<Integer, Integer> numToLocation;
        ArrayList<Integer> nums;

        public RandomizedSet() {
            numToLocation = new HashMap<>();
            nums = new ArrayList<>();
        }

        public boolean insert(int val) {
            if (numToLocation.containsKey(val)) {
                return false;
            }

            numToLocation.put(val, nums.size());
            nums.add(val);
            return true;
        }

        public boolean remove(int val) {
            if (!numToLocation.containsKey(val)) {
                return false;
            }

            int location = numToLocation.get(val);

            numToLocation.put(nums.get(nums.size() - 1), location);
            numToLocation.remove(val);
            nums.set(location, nums.get(nums.size() - 1));
            nums.remove(nums.size() - 1);
            return true;
        }

        public int getRandom() {
            Random random = new Random();
            int r = random.nextInt(nums.size());
            return nums.get(r);
        }
    }
}
相关推荐
甜瓜看代码6 分钟前
Android事件分发机制
面试
李重楼34 分钟前
前端性能优化之 HTTP/2 多路复用
前端·面试
咖啡Beans44 分钟前
使用OpenFeign实现微服务间通信
java·spring cloud
我不是混子1 小时前
说说单例模式
java
lecepin2 小时前
AI Coding 资讯 2025-09-17
前端·javascript·面试
间彧3 小时前
SimpleDateFormat既然不推荐使用,为什么java 8+中不删除此类
java
间彧3 小时前
DateTimeFormatter相比SimpleDateFormat在性能上有何差异?
java
间彧3 小时前
为什么说SimpleDateFormat是经典的线程不安全类
java
MacroZheng3 小时前
横空出世!MyBatis-Plus 同款 ES ORM 框架,用起来够优雅!
java·后端·elasticsearch