LeetCode 4.尋找兩個有序數(shù)組的中位數(shù)

題目描述:

給定兩個大小為 m 和 n 的有序數(shù)組 nums1 和 nums2。

請你找出這兩個有序數(shù)組的中位數(shù),并且要求算法的時間復(fù)雜度為 O(log(m + n))。

你可以假設(shè) nums1 和 nums2 不會同時為空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

則中位數(shù)是 2.0
示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

則中位數(shù)是 (2 + 3)/2 = 2.5

直接上代碼:

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
    int nums[5000], numsSize = nums1Size + nums2Size;
    int i = 0,j = 0,k = 0;
    
    //將兩個有序的數(shù)組合并
    for(i = 0; j < nums1Size && k < nums2Size; i++){
        if(nums1[j] > nums2[k]) {
            nums[i] = nums2[k++];
        }
        else {
            nums[i] = nums1[j++];
        }
    }
    
    //將剩余的合并
    while(j < nums1Size) 
        nums[i++] = nums1[j++];
    while(k < nums2Size) 
        nums[i++] = nums2[k++];
    
    //這里必須是2.0,因為返回的是double類型
    if(numsSize % 2 == 0) {
        return (nums[numsSize / 2] + nums[numsSize / 2 - 1]) / 2.0;
    } 
    else {
        return nums[numsSize / 2];
    }
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容