Leetcode--380. O(1) 时间插入、删除和获取随机元素

在这里插入图片描述
需要实现时间复杂度为O(1),则需要建立哈希表生成直接索引,

class RandomizedSet {
        private ArrayList<Integer> list;
        private HashMap<Integer,Integer> listMap;
        private Random random;
        private int listSize;

        public RandomizedSet() {
            list = new ArrayList<>();
            listMap = new HashMap<>();
            random = new Random();
            listSize = 0;
        }

        public boolean insert(int val) {
            if(!listMap.containsKey(val))   //如果哈希表内不存在val值对list索引的键值对
            {
                listMap.put(val,listSize);  //哈希表增加键值对,即对list的索引
                list.add(listSize,val);
                listSize++; //list大小增加
                return true;
            }
            return false;
        }

        public boolean remove(int val) {
            if(listMap.containsKey(val))
            {
                list.set(listMap.get(val),list.get(listSize-1));  //将列表最后的元素替换移除的元素
                listMap.put(list.get(listSize-1),listMap.get(val));
                list.remove(listSize-1);
                listMap.remove(val);
                listSize--;
                return true;
            }
            return false;
        }

        public int getRandom() {

            return list.get(random.nextInt(list.size()));
        }
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值