381. Insert Delete GetRandom O(1) - Duplicates allowed

public class RandomizedCollection {
    private List<Integer> nums;
    private Map<Integer,Set<Integer>> indexs;
    private Random r;
    /** Initialize your data structure here. */
    public RandomizedCollection() {
        nums=new ArrayList<Integer>();
        indexs=new HashMap<Integer,Set<Integer>>();
        r=new Random();
    }

    /** Inserts a value to the collection. Returns true if the collection did not already contain the specified element. */
    public boolean insert(int val) {
        boolean contains=indexs.containsKey(val);
        if(!contains)indexs.put(val, new LinkedHashSet<Integer>());
        indexs.get(val).add(nums.size());
        nums.add(val);
        return !contains;
    }

    /** Removes a value from the collection. Returns true if the collection contained the specified element. */
    public boolean remove(int val) {
        if(!indexs.containsKey(val))
            return false;
        int index=indexs.get(val).iterator().next();
        indexs.get(val).remove(index);
        if(index<nums.size()-1)
        {
            int lastone=nums.get(nums.size()-1);
            nums.set(index, lastone);
            indexs.get(lastone).remove(nums.size()-1);
            indexs.get(lastone).add(index);
        }
        nums.remove(nums.size()-1);
        if(indexs.get(val).isEmpty())
            indexs.remove(val);
        return true;
    }

    /** Get a random element from the collection. */
    public int getRandom() {
        return nums.get(r.nextInt(nums.size()));
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

hebastast

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值