给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n))。
示例 1:
nums1 = [1, 3] nums2 = [2] 中位数是 2.0
示例 2:
nums1 = [1, 2] nums2 = [3, 4] 中位数是 (2 + 3)/2 = 2.5
public double FindMedianSortedArrays(int[] nums1, int[] nums2) { int m = nums1.Length; int n = nums2.Length; int i = 0; int j = 0; int index = 0; int[] datas = new int[m+n]; //合并排序 while(i<m && j<n) { if(nums1[i] < nums2[j]) { datas[index++] = nums1[i++]; } else { datas[index++] = nums2[j++]; } } //填充剩下的 while(j<n) { datas[index++] = nums2[j++]; } while(i<m) { datas[index++] = nums1[i++]; } int mid = (m+n)/2; int offset = (m+n)%2; if(offset == 0) { return (datas[mid]+datas[mid-1])/2.0f; }else { return datas[mid]; } }
使用合并排序的方式,将两个数组合并排序到一个数组中,然后判断长度为偶数就取中间两个的平均数,为奇数就去取中间那位。时间复杂度应该是O(n)或O(m),空间复杂度是O(m+n)。还有更优的解法。