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

嗯 這一題還是哈希表间学,需要注意,不管存不存在都需要進行存儲添加追驴,所以主要問題在于remove部分,如果存在字典中疏之,刪除后返回true殿雪,這時候的做法,首先把同一個value的通過next(iter())取到一個坐標(biāo)值锋爪,把這個坐標(biāo)上的值和最后一個數(shù)字交換丙曙,此時字典中可以把這個坐標(biāo)刪除掉,如果這個坐標(biāo)是最后一位其骄,直接刪掉亏镰,如果不是,因為把處在最后一位上的數(shù)字交換過來了拯爽,所以最后一位上的數(shù)字現(xiàn)在存在了idx坐標(biāo)上索抓,因此需要修改這個val的最后一個坐標(biāo)值,把它變成idx,同時要刪除他原來的最后一個坐標(biāo)值纸兔。還需要判斷是不是字典中對應(yīng)的坐標(biāo)已經(jīng)被刪除完畢了惰瓜,如果刪除完畢了,要把條目也刪掉汉矿。

import collections
from random import randint 
class RandomizedCollection(object):

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.lists = []
        self.zd = collections.defaultdict(set)
        

    def insert(self, val):
        """
        Inserts a value to the collection. Returns true if the collection did not already contain the specified element.
        :type val: int
        :rtype: bool
        """
        if val in self.zd:
            flag = False
        else:
            flag = True
        self.lists.append(val)
        self.zd[val].add(len(self.lists)-1)
        return flag
           
        

    def remove(self, val):
        """
        Removes a value from the collection. Returns true if the collection contained the specified element.
        :type val: int
        :rtype: bool
        """
        if val in self.zd:
            idx = next(iter(self.zd[val]))
            self.lists[idx], self.lists[len(self.lists)-1] = self.lists[len(self.lists)-1], self.lists[idx]
            self.zd[val].remove(idx)
            if idx < len(self.lists) - 1:
                self.zd[self.lists[idx]].remove(len(self.lists)-1)
                self.zd[self.lists[idx]].add(idx)
            self.lists.pop()
            if len(self.zd[val]) == 0:
                del self.zd[val]
            return True
        else:
            return False
        

    def getRandom(self):
        """
        Get a random element from the collection.
        :rtype: int
        """
        rdx = randint( 0, len(self.lists)-1)
        return self.lists[rdx]
        


# Your RandomizedCollection object will be instantiated and called as such:
# obj = RandomizedCollection()
# param_1 = obj.insert(val)
# param_2 = obj.remove(val)
# param_3 = obj.getRandom()
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末崎坊,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子洲拇,更是在濱河造成了極大的恐慌奈揍,老刑警劉巖,帶你破解...
    沈念sama閱讀 206,126評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件赋续,死亡現(xiàn)場離奇詭異男翰,居然都是意外死亡,警方通過查閱死者的電腦和手機纽乱,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,254評論 2 382
  • 文/潘曉璐 我一進店門蛾绎,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人鸦列,你說我怎么就攤上這事租冠。” “怎么了薯嗤?”我有些...
    開封第一講書人閱讀 152,445評論 0 341
  • 文/不壞的土叔 我叫張陵顽爹,是天一觀的道長。 經(jīng)常有香客問我骆姐,道長镜粤,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 55,185評論 1 278
  • 正文 為了忘掉前任玻褪,我火速辦了婚禮肉渴,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘归园。我一直安慰自己黄虱,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 64,178評論 5 371
  • 文/花漫 我一把揭開白布庸诱。 她就那樣靜靜地躺著捻浦,像睡著了一般。 火紅的嫁衣襯著肌膚如雪桥爽。 梳的紋絲不亂的頭發(fā)上朱灿,一...
    開封第一講書人閱讀 48,970評論 1 284
  • 那天,我揣著相機與錄音钠四,去河邊找鬼盗扒。 笑死跪楞,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的侣灶。 我是一名探鬼主播甸祭,決...
    沈念sama閱讀 38,276評論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼褥影!你這毒婦竟也來了池户?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 36,927評論 0 259
  • 序言:老撾萬榮一對情侶失蹤凡怎,失蹤者是張志新(化名)和其女友劉穎校焦,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體统倒,經(jīng)...
    沈念sama閱讀 43,400評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡寨典,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 35,883評論 2 323
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了房匆。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片耸成。...
    茶點故事閱讀 37,997評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖坛缕,靈堂內(nèi)的尸體忽然破棺而出墓猎,到底是詐尸還是另有隱情,我是刑警寧澤赚楚,帶...
    沈念sama閱讀 33,646評論 4 322
  • 正文 年R本政府宣布,位于F島的核電站骗卜,受9級特大地震影響宠页,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜寇仓,卻給世界環(huán)境...
    茶點故事閱讀 39,213評論 3 307
  • 文/蒙蒙 一举户、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧遍烦,春花似錦俭嘁、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,204評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至罢猪,卻和暖如春近她,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背膳帕。 一陣腳步聲響...
    開封第一講書人閱讀 31,423評論 1 260
  • 我被黑心中介騙來泰國打工粘捎, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 45,423評論 2 352
  • 正文 我出身青樓攒磨,卻偏偏與公主長得像泳桦,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子娩缰,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 42,722評論 2 345

推薦閱讀更多精彩內(nèi)容