class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
i=0 #i遍历nums1 j遍历nums2
j=0
tmp=[]
if nums1 == []:
tmp=nums2
elif nums2 == []:
tmp=nums1
else:
while(i < len(nums1) and j < len(nums2)):
if nums1[i] < nums2[j]:
if i == len(nums1)-1:
tmp.append(nums1[i])
tmp.extend(nums2[j:])
break
tmp.append(nums1[i])
i=i+1
if nums1[i] >= nums2[j]:
if j == len(nums2)-1:
tmp.append(nums2[j])
tmp.extend(nums1[i:])
break
tmp.append(nums2[j])
j=j+1
if len(tmp) % 2 == 1: #如果是奇数
result=tmp[int(len(tmp)/2)]
else:
result=(tmp[int(len(tmp)/2)]+tmp[int(len(tmp)/2)-1])/2
return result
思路:合并排序
由于题目给出的两个数组已经有序,所以遍历两个数组,在遍历过程中对比两个数组的元素哪个小,将小的插入到新的数组中,从而得到一个排好序的合并数组
(1)令i指向nums1,j指向nums2
(2)对比i所指向的元素和j所指向的元素哪个小,将较小的元素放入tmp数组中
(3)在将较小元素的放入后,其对应的指针向后走
(4)重复(2)和(3)直到两个数组都遍历完成
(5)算出中位数,要注意长度为奇数还是偶数的区别
优化思路(代码未优化):
根据nums1和nums2的长度可以快速求出计算中位数所需要元素的位置,遍历数组时遍历到该位置即可停止,省去合并成新数组的空间复杂度以及时间复杂度