4. Median of Two Sorted Arrays
- 20min
题意:
给出两个有序数组,找出他们的中位数,时间复杂度控制在O(log(m+n))
方法一:归并做法,O(m+n)
- 时间O(M+N)
- 空间O(1)
- 两个数组长度之和为len
- 那么对于奇数个元素,中位数的下标是len/2
- 对于偶数个元素,中位数是:(len/2-1 + len/2)/2
- 所以,利用归并的思想遍历两个数组,当前元素为right,上一个元素为left
- 遍历到len/2下标后结束,若len是偶数,那么中位数是(left+right)/2,否则中位数是right
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len = nums1.length + nums2.length;
int i = 0;
int j = 0;
int left = 0;
int right = 0;
int count = 0;
while(count++ <= len / 2){
left = right;
if(i >= nums1.length){
right = nums2[j++];
}else if(j >= nums2.length){
right = nums1[i++];
}else{
right = nums1[i] <= nums2[j] ? nums1[i++] : nums2[j++];
}
}
if((len & 1) == 0){//len为偶数
return (double)(left + right) / 2;
}else{
return right;
}
}
}
- 注意:
- 判断奇偶的一个方法是
&1
,偶数返回0,奇数返回1