题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
给定两个大小为 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
Java代码:
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int totalLen = nums1.length + nums2.length;
//判断是不是奇数
boolean odd = totalLen%2 == 1;
int[] newNums = new int[totalLen];
int count = 0;
int index1 = 0;
int index2 = 0;
while(index1 < nums1.length || index2 < nums2.length){
if(index1 >= nums1.length){
newNums[count] = nums2[index2];
index2++;
}else if(index2 >= nums2.length){
newNums[count] = nums1[index1];
index1++;
}else{
int val1 = nums1[index1];
int val2 = nums2[index2];
if(val1 < val2){
newNums[count] = val1;
index1++;
}else{
newNums[count] = val2;
index2++;
}
}
if(odd && count == totalLen/2){
return newNums[count];
}
if(!odd && count == totalLen/2){
return (newNums[count-1] + newNums[count])/2.0;
}
count++;
}
return 0;
}
}
网友评论