iframe-proxy | Sunbelt Computer Software

56 lines (50 loc) · 1.82 KB

package code;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
 * 380. Insert Delete GetRandom O(1)
 * 题意：设计一个数据结构，插入，删除，随机获得一个元素 这三个操作的复杂度都为O(1)
 * 难度：Medium
 * 分类：Array, Hash Table, Design
 * 思路：List 的插入和删除都是O(1), 通过hashmap绑定来使得Get也为O(1)
 * Tips：和LRU哪个题类似 lc146
public class lc380 {
    public class RandomizedSet {
        HashMap<Integer, Integer> valToInd;
        List<Integer> list;
        int ind = 0;
        /** Initialize your data structure here. */
        public RandomizedSet() {
            valToInd = new HashMap<>();
            list = new ArrayList<>();
        /** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
        public boolean insert(int val) {
            if(valToInd.containsKey(val)) return false;
            list.add(val);
            valToInd.put(val,list.size()-1);
            return true;
        /** Removes a value from the set. Returns true if the set contained the specified element. */
        public boolean remove(int val) {
            int ind = valToInd.getOrDefault(val,-1);
            if(ind == -1) return false;
            Collections.swap(list,ind,list.size()-1);
            int swappedWith = list.get(ind);
            valToInd.put(swappedWith,ind);
            list.remove(list.size()-1);
            valToInd.remove(val);
            return true;
        /** Get a random element from the set. */
        public int getRandom() {
            int max = list.size();
            int min = 0;
            int ind = (int)(Math.random() * (max - min) + min);
            return list.get(ind);

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

lc380.java

Latest commit

History

lc380.java

File metadata and controls