leetcode 4 尋找兩個(gè)正序數(shù)組的中位數(shù)

class Solution {
public:
int getKthElement(const vector<int>& nums1, const vector<int>& nums2, int k) {
/* 主要思路:要找到第 k (k>1) 小的元素疹鳄,那么就取 pivot1 = nums1[k/2-1] 和 pivot2 = nums2[k/2-1] 進(jìn)行比較
* 這里的 "/" 表示整除
* nums1 中小于等于 pivot1 的元素有 nums1[0 .. k/2-2] 共計(jì) k/2-1 個(gè)
* nums2 中小于等于 pivot2 的元素有 nums2[0 .. k/2-2] 共計(jì) k/2-1 個(gè)
* 取 pivot = min(pivot1, pivot2)讽挟,兩個(gè)數(shù)組中小于等于 pivot 的元素共計(jì)不會超過 (k/2-1) + (k/2-1) <= k-2 個(gè)
* 這樣 pivot 本身最大也只能是第 k-1 小的元素
* 如果 pivot = pivot1介杆,那么 nums1[0 .. k/2-1] 都不可能是第 k 小的元素媚值。把這些元素全部 "刪除",剩下的作為新的 nums1 數(shù)組
* 如果 pivot = pivot2劣领,那么 nums2[0 .. k/2-1] 都不可能是第 k 小的元素睦尽。把這些元素全部 "刪除",剩下的作為新的 nums2 數(shù)組
* 由于我們 "刪除" 了一些元素(這些元素都比第 k 小的元素要形噬鳌)萍摊,因此需要修改 k 的值,減去刪除的數(shù)的個(gè)數(shù)
*/

    int m = nums1.size();
    int n = nums2.size();
    int index1 = 0, index2 = 0;     //并不真的修改數(shù)組如叼,而是用index表示參與下一                                                                           步運(yùn)算的數(shù)組的左邊界
    while (true) {
        // 邊界情況
        if (index1 == m) {      //數(shù)組一已經(jīng)為空冰木,因?yàn)椴淮嬖趎um1[m]
            return nums2[index2 + k - 1];   //這里的-1只是因?yàn)橛衝ums2[0]
        }
        if (index2 == n) {
            return nums1[index1 + k - 1];
        }
        if (k == 1) {               //找最小值時(shí)返回兩數(shù)組邊界的最小值
            return min(nums1[index1], nums2[index2]);
        }

        // 正常情況
        int newIndex1 = min(index1 + k / 2 - 1, m - 1); //若越界,則返回本數(shù)組最后                                                                                                      一個(gè)元素
        int newIndex2 = min(index2 + k / 2 - 1, n - 1);
        int pivot1 = nums1[newIndex1];  //pivot1=A[k/2-1]
        int pivot2 = nums2[newIndex2];
        if (pivot1 <= pivot2) {
            k -= newIndex1 - index1 + 1;    //k=k-k/2
            index1 = newIndex1 + 1;//將newIndex1和它前面的元素移除數(shù)組
        }
        else {
            k -= newIndex2 - index2 + 1;
            index2 = newIndex2 + 1;
        }
    }
}

double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    int totalLength = nums1.size() + nums2.size();
    if (totalLength % 2 == 1) { //m+n是奇數(shù)笼恰,返回第(m+n+1)/2個(gè)元素
        return getKthElement(nums1, nums2, (totalLength + 1) / 2);
    }
    else {
        return (getKthElement(nums1, nums2, totalLength / 2) + getKthElement(nums1, nums2, totalLength / 2 + 1)) / 2.0;
    }
}

};

作者:LeetCode-Solution
鏈接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xun-zhao-liang-ge-you-xu-shu-zu-de-zhong-wei-s-114/
來源:力扣(LeetCode)
著作權(quán)歸作者所有踊沸。商業(yè)轉(zhuǎn)載請聯(lián)系作者獲得授權(quán)囚衔,非商業(yè)轉(zhuǎn)載請注明出處。
根據(jù)中位數(shù)的定義雕沿,

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市猴仑,隨后出現(xiàn)的幾起案子审轮,更是在濱河造成了極大的恐慌,老刑警劉巖辽俗,帶你破解...
    沈念sama閱讀 217,542評論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件疾渣,死亡現(xiàn)場離奇詭異,居然都是意外死亡崖飘,警方通過查閱死者的電腦和手機(jī)榴捡,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,822評論 3 394
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來朱浴,“玉大人吊圾,你說我怎么就攤上這事『泊溃” “怎么了项乒?”我有些...
    開封第一講書人閱讀 163,912評論 0 354
  • 文/不壞的土叔 我叫張陵,是天一觀的道長梁沧。 經(jīng)常有香客問我檀何,道長,這世上最難降的妖魔是什么廷支? 我笑而不...
    開封第一講書人閱讀 58,449評論 1 293
  • 正文 為了忘掉前任频鉴,我火速辦了婚禮,結(jié)果婚禮上恋拍,老公的妹妹穿的比我還像新娘垛孔。我一直安慰自己,他們只是感情好芝囤,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,500評論 6 392
  • 文/花漫 我一把揭開白布似炎。 她就那樣靜靜地躺著,像睡著了一般悯姊。 火紅的嫁衣襯著肌膚如雪羡藐。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,370評論 1 302
  • 那天悯许,我揣著相機(jī)與錄音仆嗦,去河邊找鬼。 笑死先壕,一個(gè)胖子當(dāng)著我的面吹牛瘩扼,可吹牛的內(nèi)容都是我干的谆甜。 我是一名探鬼主播,決...
    沈念sama閱讀 40,193評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼集绰,長吁一口氣:“原來是場噩夢啊……” “哼规辱!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起栽燕,我...
    開封第一講書人閱讀 39,074評論 0 276
  • 序言:老撾萬榮一對情侶失蹤罕袋,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后碍岔,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體浴讯,經(jīng)...
    沈念sama閱讀 45,505評論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,722評論 3 335
  • 正文 我和宋清朗相戀三年蔼啦,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了榆纽。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,841評論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡捏肢,死狀恐怖奈籽,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情鸵赫,我是刑警寧澤唠摹,帶...
    沈念sama閱讀 35,569評論 5 345
  • 正文 年R本政府宣布,位于F島的核電站奉瘤,受9級特大地震影響勾拉,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜盗温,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,168評論 3 328
  • 文/蒙蒙 一藕赞、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧卖局,春花似錦斧蜕、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,783評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至染坯,卻和暖如春均芽,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背单鹿。 一陣腳步聲響...
    開封第一講書人閱讀 32,918評論 1 269
  • 我被黑心中介騙來泰國打工掀宋, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 47,962評論 2 370
  • 正文 我出身青樓劲妙,卻偏偏與公主長得像湃鹊,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個(gè)殘疾皇子镣奋,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,781評論 2 354