leetcode Median of Two Sorted Arrays python

对于两个已排序的数组,在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 :
                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


你可能感兴趣的:(python)