Java Map与List的应用

Java Map与List的应用

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/insert-delete-getrandom-o1

实现RandomizedSet 类：

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

class RandomizedSet {
   private List<Integer> list ;
   private Map<Integer,Integer> map;
   private Random random;
    public RandomizedSet() {
        list = new ArrayList<Integer>();
        map = new HashMap<Integer,Integer>();
        random = new Random();
    }
    
    public boolean insert(int val) {
        if(list.contains(val)) return false;
        map.put(val,list.size());
        return list.add(val);
    }
    
    /**
    这道题考的是数据结构的理解,List无法针对值删除数据，Map无法随机查询
    单独的数据结构无法满足值删除及随机读的功能，所以删除时需要借助Map存储
    index进行对值的删除,list对索引做随机查询
     */
    public boolean remove(int val) {
        if(!list.contains(val)) return false;
        int index = map.get(val);
        int box = list.size();
        int last = list.get(box-1);
        list.set(index,last);
        list.remove(box-1);
        map.put(last,index);
        map.remove(val);
        return true;
    }
    
    public int getRandom() {
        int box = list.size();
        int val = list.get(random.nextInt(box));
        return val;
    }
}