力扣4：寻找两个正序数的中位数

一个不喜欢and不会代码的码农2024-10-25 5:02

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

复制代码

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：

复制代码

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

思想：先按照归并排序进行合并两个有序表。然后按照合并后的顺序表偶数个和奇数个数据分别进行寻找中位数。

代码：

复制代码

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
	int *nums=(int*)malloc(sizeof(int)*(nums1Size+nums2Size));
	int i=0,j=0,k=0;
	while(i<=nums1Size-1&&j<=nums2Size-1){
		if(nums1[i]<nums2[j]){
			nums[k++]=nums1[i++];
		}else{
			nums[k++]=nums2[j++];
		}
	}
	while(i<=nums1Size-1) nums[k++]=nums1[i++];
	while(j<=nums2Size-1) nums[k++]=nums2[j++];
	if((nums1Size+nums2Size)%2==0){
        //数组下标从0开始
		return ((double)nums[(nums1Size+nums2Size)/2]+(double)nums[(nums1Size+nums2Size)/2-1])/2;
	}else{
		 return (double)nums[(nums1Size+nums2Size)/2];
	}
}

时间复杂度O（log(m+n)），空间复杂度O（1）