380. Insert Delete GetRandom O(1)

Medium
這個面經題一做,暴露出來不少基礎問題。比如hashmap的key是不能修改的,只能remove舊的再put新的竞思,不可以modify. 還有hashMap的key, value一定要給自己多一點思路空間,就像之前面google那位面試官說的tree里的children做key, parent做value這種逆向思維可以多嘗試钞护,思路打開一點.

我一開始remove做成了O(n), 導致了TLE. 因為hashmap的key是index, value是val, 這樣的話要刪除某個val我得先遍歷keySet()來找該val對應的key. 后來直接用val做key的話remove時直接O(1)就找到這個val. 同時開了一個list來存vals.這樣可以很容易拿到最后insert的val, 在remove時我們是通過把最后加入的移到被刪除的地方盖喷,然后刪掉最后加入的原來的key-value pair的, 所以能方便拿到最后加入的元素很重要。同時注意一下list.remove(int index)list.remove(Object o)這兩個方法當Object是Integer的時候注意一下轉換類型难咕,不然會被當成是index.

class RandomizedSet {
    Map<Integer, Integer> map;
    List<Integer> list;
    /** Initialize your data structure here. */
    public RandomizedSet() {
        map = 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 (map.containsKey(val)){
            return false;
        } else {
            map.put(val, list.size());
            list.add(val);
            return true;
        }   
    }
    
    /** Removes a value from the set. Returns true if the set contained the specified element. */
    public boolean remove(int val) {
        if (!map.containsKey(val)){
            return false;
        } else {
            map.put(list.get(list.size() - 1), map.get(val));   
            map.remove(val);
            list.remove((Integer) val);
            return true;
        }
    }
    
    /** Get a random element from the set. */
    public int getRandom() {
        Random rand = new Random();
        int randIndx = rand.nextInt(list.size());
        return list.get(randIndx);
    }
}

/**
 * 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();
 */
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末课梳,一起剝皮案震驚了整個濱河市距辆,隨后出現的幾起案子,更是在濱河造成了極大的恐慌暮刃,老刑警劉巖跨算,帶你破解...
    沈念sama閱讀 217,084評論 6 503
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現場離奇詭異椭懊,居然都是意外死亡诸蚕,警方通過查閱死者的電腦和手機,發(fā)現死者居然都...
    沈念sama閱讀 92,623評論 3 392
  • 文/潘曉璐 我一進店門氧猬,熙熙樓的掌柜王于貴愁眉苦臉地迎上來背犯,“玉大人,你說我怎么就攤上這事盅抚∧海” “怎么了?”我有些...
    開封第一講書人閱讀 163,450評論 0 353
  • 文/不壞的土叔 我叫張陵妄均,是天一觀的道長柱锹。 經常有香客問我,道長丛晦,這世上最難降的妖魔是什么奕纫? 我笑而不...
    開封第一講書人閱讀 58,322評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮烫沙,結果婚禮上,老公的妹妹穿的比我還像新娘隙笆。我一直安慰自己锌蓄,他們只是感情好,可當我...
    茶點故事閱讀 67,370評論 6 390
  • 文/花漫 我一把揭開白布撑柔。 她就那樣靜靜地躺著瘸爽,像睡著了一般。 火紅的嫁衣襯著肌膚如雪铅忿。 梳的紋絲不亂的頭發(fā)上剪决,一...
    開封第一講書人閱讀 51,274評論 1 300
  • 那天,我揣著相機與錄音檀训,去河邊找鬼柑潦。 笑死,一個胖子當著我的面吹牛峻凫,可吹牛的內容都是我干的渗鬼。 我是一名探鬼主播,決...
    沈念sama閱讀 40,126評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼荧琼,長吁一口氣:“原來是場噩夢啊……” “哼譬胎!你這毒婦竟也來了差牛?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 38,980評論 0 275
  • 序言:老撾萬榮一對情侶失蹤堰乔,失蹤者是張志新(化名)和其女友劉穎偏化,沒想到半個月后,有當地人在樹林里發(fā)現了一具尸體镐侯,經...
    沈念sama閱讀 45,414評論 1 313
  • 正文 獨居荒郊野嶺守林人離奇死亡夹孔,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,599評論 3 334
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現自己被綠了析孽。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片搭伤。...
    茶點故事閱讀 39,773評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖袜瞬,靈堂內的尸體忽然破棺而出怜俐,到底是詐尸還是另有隱情,我是刑警寧澤邓尤,帶...
    沈念sama閱讀 35,470評論 5 344
  • 正文 年R本政府宣布拍鲤,位于F島的核電站,受9級特大地震影響汞扎,放射性物質發(fā)生泄漏季稳。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,080評論 3 327
  • 文/蒙蒙 一澈魄、第九天 我趴在偏房一處隱蔽的房頂上張望景鼠。 院中可真熱鬧,春花似錦痹扇、人聲如沸铛漓。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,713評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽浓恶。三九已至,卻和暖如春结笨,著一層夾襖步出監(jiān)牢的瞬間包晰,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,852評論 1 269
  • 我被黑心中介騙來泰國打工炕吸, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留伐憾,地道東北人。 一個月前我還...
    沈念sama閱讀 47,865評論 2 370
  • 正文 我出身青樓算途,卻偏偏與公主長得像塞耕,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子嘴瓤,可洞房花燭夜當晚...
    茶點故事閱讀 44,689評論 2 354

推薦閱讀更多精彩內容