小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。
前言
- LeetCode第四题:寻找两个正序数组的中位数,难道:困难
- 题目描述:给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请找出并返回这两个正序数组的中位数;
- 这道题给我的感觉其实还没有上次做的-无重复最长字符串-难,基本上提交一次就成功了【狗头/滑稽】;
方式一:并归数组
- 解题重点两个正序数组,所以提供的数组已经是排好序的,只需要在合并到一起凑成一个大数组即可
public double findMedianSortedArrays (int[] nums1, int[] nums2) {
int all = nums1.length + nums2.length
int[] arr = new int[all]
int i = 0, i1 = 0, i2 = 0
while (i1 < nums1.length && i2 < nums2.length) {
if (nums1[i1] <= nums2[i2]) {
arr[i++] = nums1[i1++]
} else {
arr[i++] = nums2[i2++]
}
}
while (i1 < nums1.length) {
arr[i++] = nums1[i1++]
}
while (i2 < nums2.length) {
arr[i++] = nums2[i2++]
}
int a = all >> 1
if (all % 2 == 0) {
return (arr[a - 1] + arr[a]) / 2d
}
return arr[a]
}
- 这种方式是时间复杂度是O(m+n),空间复杂度也是O(m+n);
方式二:查找中位数
- 感觉比上述的并归方式稍微难写(并归方式一下搞定通过,这种方式思考了很久才做出来的 O.O)
- 这种方式通过看官方解题介绍里有描述,看了一些其他的解题思路,还是摸索出来了。
public double findMedianSortedArrays (int[] nums1, int[] nums2) {
int len = nums1.length + nums2.length
int i = 0, i1 = 0, i2 = 0, s = len >> 1, value1, value2 = 0
do {
value1 = value2
if (i1 < nums1.length && (i2 >= nums2.length || nums1[i1] <= nums2[i2])) {
value2 = nums1[i1++]
} else {
value2 = nums2[i2++]
}
} while (s >= ++i)
return len % 2 == 0 ? (value1 + value2) / 2d : value2
}
- 这种方式是时间复杂度可以当做是O(m+n),空间复杂度是O(1);
最后
- 当然这题还有其他的解法,在其官方介绍里面使用二分查找可以使时间复杂度降低至O(log(m+n));
- 虚心学习,共同进步 -_-