码迷,mamicode.com
首页 > 编程语言 > 详细

LeetCode刷题:第四题 寻找两个有序数组的中位数

时间:2019-03-14 12:01:01      阅读:160      评论:0      收藏:0      [点我收藏+]

标签:数组   算法   有序数组   题目   假设   ade   会同   sorted   ++   

题目描述:

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2

请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

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

则中位数是 2.0

示例 2:

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

则中位数是 (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;
    
    //将两个有序的数组合并
    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];
    }
}

注释已经给出完整解释!

LeetCode刷题:第四题 寻找两个有序数组的中位数

标签:数组   算法   有序数组   题目   假设   ade   会同   sorted   ++   

原文地址:https://www.cnblogs.com/lihanwen/p/10529227.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!