本文已参与「新人创作礼」活动，一起开启掘金创作之路。
本题考察数组相关知识，为困难题4. 寻找两个正序数组的中位数
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
代码
class Solution:
def findMedianSortedArrays(self, nums1, nums2):
nums1.extend(nums2)
nums1.sort()
if len(nums1)%2 == 0:
i = int(len(nums1) / 2)
return (nums1[i-1] + nums1[i]) / 2
else:
i = int((len(nums1) + 1) / 2)
return nums1[i-1]
不用sort()
class Solution:
def findMedianSortedArrays(self, nums1, nums2):
len1 = len(nums1)
len2 = len(nums2)
i = j = 0
result = []
while (True) :
if i < len1 and j < len2:
if nums1[i] <= nums2[j]:
result.append(nums1[i])
i += 1
else:
result.append(nums2[j])
j += 1
elif i >= len1 and j < len2:
result.append(nums2[j])
j += 1
elif i < len1 and j >= len2:
result.append(nums1[i])
i += 1
else:
break
if len(result) % 2 == 0:
i = int(len(result) / 2)
return (result[i-1] + result[i]) / 2
else:
i = int((len(result) +1 ) / 2)
return result[i-1]
解析
按照我的简单至上的思路，直接放一起然后用sort排序，然后分成奇数个和偶数个两种情况按照定义找中位数。但是很显然这样写毫无难度，应该也不是想让我这么解的吧。稍微难一点的就是不用sort排序了吧。
具体方法：
1.因为两个原数列是排好序的，所以用while循环来进行排序就行了
2.然后还是按照奇数个数时取中间的，偶数个数时取中间两个的平均值即可，按照要求找到数的位置
3.输出在这个位置的数即可
来源：力扣（LeetCode） 链接：leetcode.cn/problems/me…
