题目
给定两个大小为 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
解题
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
if(nums1.length == 0 || nums2.length == 0){
int[] num = nums1.length == 0 ? nums2: nums1;
return num.length % 2 == 0 ?(num[num.length / 2] + num[(num.length / 2) - 1]) / 2.0 :num[num.length / 2];
}
int length = nums1.length + nums2.length;
int[] num = new int[length];
int k1 = 0;
int k2 = 0;
while(k1 < nums1.length || k2 < nums2.length){
if(k1 == nums1.length){
num[k1+k2] = nums2[k2];
k2++;
continue;
}
if(k2 == nums2.length){
num[k1+k2] = nums1[k1];
k1++;
continue;
}
if(nums1[k1] <= nums2[k2]){
num[k1+k2] = nums1[k1];
k1++;
}else{
num[k1+k2] = nums2[k2];
k2++;
}
}
return num.length % 2 == 0 ?(num[num.length / 2] + num[(num.length / 2) - 1]) / 2.0 :num[num.length / 2];
}
}
备注
由于这道题的时间复杂度要求为O(log(m + n)),目前还没有想到解题思路,会在继续更新这道题,望原谅