Q380 O(1)时间获取插入删除元素

news2024/9/19 10:45:22

思路

insert 其实用得到search，remove也是，当时o(1)想到的是hash set，但是对于random取,随机数相当于获得的是index,根据index获取元素 Set 数据结构不符合。

随机获取应该是数组，但是数组搜索和删除不是o(1)

最后的思路是用hash和数组存，对于数组删除，可以使用交换的方式

在这里插入图片描述

通过交换的方式进行删除。

代码

class RandomizedSet {

    private HashMap<Integer, Integer> hashMap;
    private List<Integer> list ;
    Random random = new Random();
    public RandomizedSet() {
        hashMap = new HashMap<>();
        list = new ArrayList<>();
    }


    public boolean insert(int val) {
       if (hashMap.containsKey(val)){
           return false;
       }
       int size = list.size();
       hashMap.put(val,size);
       list.add(val);
       return true;
    }

    public boolean remove(int val) {
        if (hashMap.containsKey(val)){
            int index = hashMap.get(val);
            int lastElement = list.get(list.size() - 1);
            list.set(index, lastElement);//index 放的是最后一个值
            list.remove(list.size() - 1);//把最后一个删除是o(1)
            hashMap.put(lastElement, index); // 更新最后一个元素在 hashMap 中的位置
            hashMap.remove(val);
            return true;
        }
        return false;
    }

    public int getRandom() {
        int i = random.nextInt(list.size());
        return list.get(i);
    }
}

/**
 * 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();
 */

遇到问题

Java 8在线文档
https://docs.oracle.com/javase/8/docs/api/?xd_co_f=47c934d9-e663-4eba-819c-b726fc2d0847
Java 22在线文档
https://docs.oracle.com/en/java/javase/22/docs/api/java.base/java/util/ArrayList.html#removeLast()

代码报错，是因为我使用了removeList,这个功能是Java jdk21以后才有的，我刚开始写代码用的jdk22,所以有这个提示，但是当我用Java 8 运行时，报错
在这里插入图片描述