//给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。 // // 请你找出这两个正序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。 // // 你可以假设 nums1 和 nums2 不会同时为空。
初看这道题,寻找中位数,在一个有序数组中寻找一个数第一个想到的当然是二分查找
,可以做到O(logn)
是时间复杂度。那个对于两个有序数组怎么处理呢?
第一个想到的就是将两个数组合并成一个有序数组,这样就可以直接取合并后数组的第len+1 / 2
个值(长度是奇数)或者是第len+1 / 2
和len+1 / 2 - 1
的平均值(长度是偶数);
而合并两个有序数组,显然是归并排序
的一个步骤,时间复杂度是O(m+n)
。思路有了,代码如下
package com.tx.leetcode.editor.cn;
//Java:寻找两个正序数组的中位数
public class P4MedianOfTwoSortedArrays{
public static void main(String[] args) {
int[] nums1 = {1,4};
int[] nums2 = {2,3};
double result =findMedianSortedArrays(nums1, nums2);
System.out.println(result);
}
//leetcode submit region begin(Prohibit modification and deletion)
public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.length;
int n = nums2.length;
int len = m + n;
int left = -1, right = -1;
int aStart = 0, bStart = 0;
for (int i = 0; i <= len / 2; i++) {
left = right;
if (aStart < m && (bStart >= n || nums1[aStart] < nums2[bStart])) {
right = nums1[aStart++];
} else {
right = nums2[bStart++];
}
}
if ((len & 1) == 0) { // 判断奇偶
return (left + right) / 2.0;
//
// 比如:11 转成二进制
// 1 0 1 1
// & 0 0 0 1
// ------------------
// 0 0 0 1
// 所以,奇数与1作与运算,结果肯定是1
// 以后记下来,与1相与为0,则为偶数
}
else {
return right;
}
}
}