今天看到了一道题(来自力扣)
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。
请你找出并返回这两个正序数组的 中位数 。
示例 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
-106 <= nums1[i], nums2[i] <= 106
乍一看还挺简单,然后就上手开始做了,
第一次做出来的代码如下
fn = function (nums1, nums2) {
let newnum = nums1.concat(nums2).sort();
length = newnum.length;
console.log(newnum);
if (length % 2 === 0) {
return (newnum[length / 2] + newnum[length / 2 - 1]) / 2;
} else {
return newnum[(length - 1) / 2];
}
};
console.log(fn([2], [-1, -2]));
然后上交,果不其然没通过,原因是负数没过,然后我仔细地想了一下原因,突然想起来,js的sort()算法是有缺陷的他是通过比较类似字符串的代码,逐位进行比较。
举个栗子,这个算法,2就比11大,因为,他是先比较第一位,第一位的2大于了11的首位1,后面就不再进行比较,所以这种大于10的数字和负数根本没法比较。
所以我又想得用其他排序方法,然后就写上了最熟悉的冒泡排序,以为会超时,结果发现居然通过了,也是很意外。
后来又想了想,关于这个sort()算法,想要排序还真有一种方法,sort((a, b) => a - b)参数是一个回调函数, 回调函数有2个参数, 分别是当前数组元素与其前一个数组元素, 也就是你代码里的a 和 b, 数组的返回值决定了是否要交换 a 和 b 在数组中的顺序, 返回值小于0就表示要交换 a 和 b 在数组中的顺序, 否则不交换顺序。
后来代码又改成了
fn = function (nums1, nums2) {
let newnum = nums1.concat(nums2).sort((a, b) => a - b);
length = newnum.length;
console.log(newnum);
if (length % 2 === 0) {
return (newnum[length / 2] + newnum[length / 2 - 1]) / 2;
} else {
return newnum[(length - 1) / 2];
}
};
console.log(fn([2], [-1, -2]));
注:nums1.concat(nums2)和[...nums1,...nums2]都可以把两个数组组成一个数组,前者是在nums1后面直接追加,后者是生成一个新的数组作为返回值。