对于两个已排序的数组,在o(n+m)的时间内就可以按序存储在一个大数组里面,这里学习python语法,记录代码.
class Solution:
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
l1 = len(nums1)
l2 = len(nums2)
a = 0
b = 0
nums3 = []
while a<l1 or b<l2:
if a>=l1 :
nums3.append(nums2[b])
b+=1
continue
if b>=l2:
nums3.append(nums1[a])
a+=1
continue
if nums1[a]<=nums2[b] :
nums3.append(nums1[a])
a+=1
else :
nums3.append(nums2[b])
b+=1
g = (l1+l2+1)//2-1
if (l1+l2)&1==1:
return nums3[g]*1.0
return (nums3[g]+nums3[g+1])*1.0/2.0