给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
拿到题目,一如既往的解题思路:将两个数组合并在一起,然后排序,判断该数组长度是奇数还是偶数,在找中间的数。但这似乎因为python内置的一些库和语言,反而很容易实现,耗时28ms,15.1mb:
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
nums = sorted(nums1 + nums2)
idex1 = len(nums)//2
if len(nums)%2 == 0:
idex2 = len(nums)//2 - 1
return ((nums)[idex1]+(nums)[idex2])/2
else:
return (nums)[idex1]
当然还有另一种角度,不用内置的sorted函数,给两个数组都赋一个指针,寻找中位数,就是寻找第k小的值。这种方法写起来较为复杂,不如上面简便。
注意:上述代码用的是sorted(),而不是sort()。
二者的区别是:
sort()无返回值的,直接改变原有的列表;
sorted()是返回一个排序好的列表,它并不会修改原有的列表。
python中的sort使用的是一种名为Timsort的排序方法,其时间复杂度为O(n log n),是一种快速的稳定的排序方法。具体参考https://blog.csdn.net/u010883226/article/details/84403263