leetcode -- 寻找两个有序数组的中位数
题目:
给定两个大小为 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int[] a = new int[nums1.length+nums2.length];
int i=0,j=0,k=0;
while(i<nums1.length && j<nums2.length){
if(nums1[i]<=nums2[j]){
a[k++] = nums1[i++];
}else{
a[k++] = nums2[j++];
}
}
while(i<nums1.length){
a[k++] = nums1[i++];
}
while(j<nums2.length){
a[k++] = nums2[j++];
}
double x = 0.0;
int mid = (nums1.length+nums2.length)/2;
if((nums1.length+nums2.length)%2==0){
x = Double.parseDouble((a[mid-1]+a[mid])+"")/2;
}else{
x = a[mid];
}
return x;
}
}