题目:
https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
4. 寻找两个正序数组的中位数
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。
请你找出并返回这两个正序数组的中位数。
进阶:你能设计一个时间复杂度为 O(log(m+n)) 的算法解决此问题吗?
示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
示例 4:
输入:nums1 = [], nums2 = [1]
输出:1.00000
示例 5:
输入:nums1 = [2], nums2 = []
输出:2.00000
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-10^6 <= nums1[i], nums2[i] <= 10^6
解:
笨比行为大赏:
首先是最容易想到的,把两个数组合并起来,然后取中位数。这个就忽略了。
总长度固定,那么要找的中位数的索引也能确定下来,然后用两个指针分别指向两个数组的位置,一个一个往后找,循环特定的次数,然后总个数分别为奇偶数做不同的处理,最后取数,就得到了结果,这个方法的复杂度应该是O(m+n)。空间复杂度为O(1)。
publicstatic double findMedianSortedArrays(int[] nums1, int[] nums2) {
double result = 0;
int length = nums1.length + nums2.length;
int point1 = 0;
int point2 = 0;
if (length % 2 == 0) {
// 总数为偶数时
for (int i = 0; i < length / 2 - 1; i++) {
if (point2 == nums2.length) {
point1++;
} else if (point1 == nums1.length) {
point2++;
} else if (nums2[point2] <= nums1[point1]) {
point2++;
} else if (nums1[point1] < nums2[point2]) {
point1++;
}
}
for (int i = 0; i < 2; i++) {
if (point2 == nums2.length) {
result += nums1[point1];
point1++;
} else if (point1 == nums1.length) {
result += nums2[point2];
point2++;
} else if (nums1[point1] <= nums2[point2]) {
result += nums1[point1];
point1++;
} else if (nums2[point2] < nums1[point1]) {
result += nums2[point2];
point2++;
}
}
result = result / 2;
} else {
// 总数为奇数时
for (int i = 0; i < length / 2; i++) {
if (point2 == nums2.length) {
point1++;
} else if (point1 == nums1.length) {
point2++;
} else if (nums2[point2] <= nums1[point1]) {
point2++;
} else if (nums1[point1] < nums2[point2]) {
point1++;
}
}
if (point2 == nums2.length) {
result += nums1[point1];
point1++;
} else if (point1 == nums1.length) {
result += nums2[point2];
point2++;
} else if (nums1[point1] <= nums2[point2]) {
result += nums1[point1];
point1++;
} else if (nums2[point2] < nums1[point1]) {
result += nums2[point2];
point2++;
}
}
return result;
}
非常笨的方法,没想到用时还不是倒数,中间要注意数组越界和大于小于号的问题,其他的就没啥了。
但是题目中很明显的给出了要求为时间复杂度O(log(m+n)),基本可以确定要用二分法来实现。
看了看题解,放弃。
Over!