题目描述:
给定两个大小为 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) {
int l1 = nums1.length;
int l2 = nums2.length;
if (l1==0 || l2 ==0) {
return l1==0?(l2%2==0?((nums2[l2/2-1]+nums2[l2/2])*1.0/2):(nums2[(l2-1)/2])):
(l1%2==0?((nums1[l1/2-1]+nums1[l1/2])*1.0/2):(nums1[(l1-1)/2]));
}
int[] newNums = new int[l1+l2];
int index = 0;
int y=0;
for (int num:nums1) {
for (;y<l2;y++) {
if (num>nums2[y]) {
newNums[index] = nums2[y];
index++;
} else {
newNums[index] = num;
index++;
break;
}
}
if(y>=l2) {
newNums[index] = num;
index++;
}
}
for (;y<l2;y++) {
if (newNums[index-1]<=nums2[y]) {
newNums[index] = nums2[y];
index++;
}
}
int ll = newNums.length/2;
return newNums.length%2==0?(newNums[ll]+newNums[ll-1])*1.0/2:newNums[ll];
}
}
结语
运行时间可以满足题目要求,但是代码略冗余,记录一下,慢慢debug,毕竟你不写bug,就得改bug.