给定两个大小为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
ac:100%
import java.util.ArrayList;
public class LeetCode04寻找两个有序数组的中位数 {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
ArrayList<Integer> list = new ArrayList<Integer>();
int i = 0, j = 0;
while(i < nums1.length || j < nums2.length){
if(i < nums1.length && j < nums2.length){
if(nums1[i] < nums2[j]){
list.add(nums1[i]);
i++;
} else {
list.add((nums2[j]));
j++;
}
} else if(i < nums1.length) {
list.add(nums1[i]);
i++;
} else {
list.add((nums2[j]));
j++;
}
}
if(list.size() % 2 == 0){
int center = list.size() / 2;
return (double) (list.get(center) + list.get(center - 1)) / 2;
} else {
return list.get(list.size() / 2);
}
}
}