6.21 - hard - 1

4. Median of Two Sorted Arrays

這題基本想法是扔一半,比如說取第k大的數(shù)秸讹,那么則比較nums1[k/2-1] 和nums2[k/2-1] 例如 nums1[k/2]這個值比較大,則說明第k大的數(shù)肯定不再nums2的 0 ~ k/2-1的這個范圍內(nèi)稠通,全部扔掉钮科。這時候再遞歸的時候nums2 從 0 + k/2開始計數(shù),而k的值也相應(yīng)的縮小為 k - k/2彼乌。最tricky的地方泻肯,也是我花了一些時間的地方,就是到底從哪里開始計數(shù)囤攀,扔掉多少软免,新的pointer更新為多少,k一定要從1開始count焚挠,否則不太好做膏萧。

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        # 和kth largest number in two sorted array 類似
        l = len(nums1) + len(nums2)
        if l % 2 != 0:
            return self.kth(nums1, nums2, 0, 0, l/2+1)
        else:
            return (self.kth(nums1, nums2, 0, 0, l/2) + self.kth(nums1, nums2, 0, 0, l/2+1)) / 2.0
    
    def kth(self, nums1, nums2, p1, p2, k): 
        """ find kth element """
        # k is kth element counting from 1
        if p1 >= len(nums1):
            return nums2[p2+k-1]
        if p2 >= len(nums2):
            return nums1[p1+k-1]
        
        if k == 1:
            return min(nums1[p1], nums2[p2])
        
        if p1+k/2-1 >= len(nums1):
            cur1 = sys.maxint
        else:
            cur1 = nums1[p1+k/2-1]
        
        if p2+k/2-1 >= len(nums2):
            cur2 = sys.maxint
        else:
            cur2 = nums2[p2+k/2-1]
        
        if cur1 > cur2:
            return self.kth(nums1, nums2, p1, p2+k/2, k - k/2)
        else:
            return self.kth(nums1, nums2, p1+k/2, p2, k - k/2)
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市蝌衔,隨后出現(xiàn)的幾起案子榛泛,更是在濱河造成了極大的恐慌,老刑警劉巖噩斟,帶你破解...
    沈念sama閱讀 222,946評論 6 518
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件曹锨,死亡現(xiàn)場離奇詭異,居然都是意外死亡剃允,警方通過查閱死者的電腦和手機沛简,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,336評論 3 399
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來斥废,“玉大人椒楣,你說我怎么就攤上這事∧等猓” “怎么了捧灰?”我有些...
    開封第一講書人閱讀 169,716評論 0 364
  • 文/不壞的土叔 我叫張陵,是天一觀的道長统锤。 經(jīng)常有香客問我毛俏,道長,這世上最難降的妖魔是什么饲窿? 我笑而不...
    開封第一講書人閱讀 60,222評論 1 300
  • 正文 為了忘掉前任煌寇,我火速辦了婚禮,結(jié)果婚禮上逾雄,老公的妹妹穿的比我還像新娘阀溶。我一直安慰自己,他們只是感情好嘲驾,可當我...
    茶點故事閱讀 69,223評論 6 398
  • 文/花漫 我一把揭開白布淌哟。 她就那樣靜靜地躺著迹卢,像睡著了一般辽故。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上腐碱,一...
    開封第一講書人閱讀 52,807評論 1 314
  • 那天誊垢,我揣著相機與錄音掉弛,去河邊找鬼。 笑死喂走,一個胖子當著我的面吹牛殃饿,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播芋肠,決...
    沈念sama閱讀 41,235評論 3 424
  • 文/蒼蘭香墨 我猛地睜開眼乎芳,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了帖池?” 一聲冷哼從身側(cè)響起奈惑,我...
    開封第一講書人閱讀 40,189評論 0 277
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎睡汹,沒想到半個月后肴甸,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,712評論 1 320
  • 正文 獨居荒郊野嶺守林人離奇死亡囚巴,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,775評論 3 343
  • 正文 我和宋清朗相戀三年原在,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片彤叉。...
    茶點故事閱讀 40,926評論 1 353
  • 序言:一個原本活蹦亂跳的男人離奇死亡庶柿,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出姆坚,到底是詐尸還是另有隱情澳泵,我是刑警寧澤,帶...
    沈念sama閱讀 36,580評論 5 351
  • 正文 年R本政府宣布兼呵,位于F島的核電站兔辅,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏击喂。R本人自食惡果不足惜维苔,卻給世界環(huán)境...
    茶點故事閱讀 42,259評論 3 336
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望懂昂。 院中可真熱鬧介时,春花似錦、人聲如沸凌彬。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,750評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽铲敛。三九已至褐澎,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間伐蒋,已是汗流浹背工三。 一陣腳步聲響...
    開封第一講書人閱讀 33,867評論 1 274
  • 我被黑心中介騙來泰國打工迁酸, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人俭正。 一個月前我還...
    沈念sama閱讀 49,368評論 3 379
  • 正文 我出身青樓奸鬓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親掸读。 傳聞我的和親對象是個殘疾皇子串远,可洞房花燭夜當晚...
    茶點故事閱讀 45,930評論 2 361

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