厚积薄发打卡Day96：哈希表（一）＜实现查删插为O(1)的容器＞

本文链接：https://blog.csdn.net/qq_42191980/article/details/121199285

哈希表：

哈希表是一种常见的数据结构，其最大有点是高效，在哈希表的插入、删除或查找元素的时候只需要O(1)的时间，因此哈希表常被用来优化时间效率。

在Java中，哈希表有两个对应的类型，即HashSet和HashMap：

HashMap：映射关系
- 结构：数组+链表+红黑树（1.8+）
- key为set结构：不可重复
- value为collection结构：可重复
HashSet：集合
- 结构：HashMap的key

题目一：

设计一个数据结构，使得如下三个操作的时间复杂度都为O(1)：

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

思路：

题目中要求需要用到O(1)的数据结构，HashMap必选（可实现插入和删除的要求），但光用map结构无法满足等概率返回每个数值，因为没法保证hash桶里面的元素都相等，因此可以借助数组（可以实现在一个长度为N的数组中，每个元素的概率都是均等为为1/N）：所以运用 HashMap+Arraylist实现
- 数组存储数据直接在后面添加（O(1)），但是删除元素时需要遍历找到对应元素才能进行对症下药，因此我们需要用映射，对数据进行一阵见血的查找（~~成语鬼才了属于是~~）
插入：
- 用map键值记录表校验，如果存在则直接return false
- 不存在则先在map记录（val，val所在数组的下标）
- list直接添加数据
删除：
- 用map键值记录表校验，如果不存在则直接return false
- 存在则获取 val 数组所在下标，对于map记录表直接remove即可
- 对于不能确定是否在数组最后一位的元素，直接可能会造成数组的移动，这就会造成O(n)的时间复杂度
- 因此需要将数组和最后一个元素进行互换，再进行删除即可
- 在此之前需要更新map键值记录表校验
获取随机数：
- 获取在数组长度的任意数字即可

实现：

public 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);
        //删除val在键值记录表的数据
        numToLocation.remove(val);
        //容器交换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);
    }
}

验证：

public static void main(String[] args) {
    RandomizedSet randomizedSet = new RandomizedSet();
    randomizedSet.insert(10);
    randomizedSet.insert(8);
    randomizedSet.insert(6);
    randomizedSet.insert(4);
    randomizedSet.insert(2);
    randomizedSet.insert(3);

    randomizedSet.remove(4);
    randomizedSet.remove(8);

    int random = randomizedSet.getRandom();
    System.out.println(random);
}