4.两个排序数组的中位数

# 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
# 请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
# 你可以假设 nums1 和 nums2 不同时为空。

# 示例：
# nums1 = [1, 3]
# nums2 = [2]
# 中位数是 2.0

# 难度：困难


class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        var = len(nums1) + len(nums2)
        list1 = []
        a = 0
        b = 0
        if var % 2 == 0:
            for i in range(1, int(var / 2 + 2)):
                if a < len(nums1) and b < len(nums2):
                    if nums1[a] < nums2[b]:
                        list1.append(nums1[a])
                        a += 1
                    else:
                        list1.append(nums2[b])
                        b += 1
                elif a >= len(nums1):
                    list1.append(nums2[b])
                    b += 1
                else:
                    list1.append(nums1[a])
                    a += 1
            return (list1[int(var / 2)] + list1[int(var / 2 - 1)]) / 2
        else:
            for i in range(1, int((var + 1) / 2) + 1):
                if a < len(nums1) and b < len(nums2):
                    if nums1[a] < nums2[b]:
                        list1.append(nums1[a])
                        a += 1
                    else:
                        list1.append(nums2[b])
                        b += 1
                elif a >= len(nums1):
                    list1.append(nums2[b])
                    b += 1
                else:
                    list1.append(nums1[a])
                    a += 1
            return list1[int((var + 1) / 2) - 1]


print(Solution.findMedianSortedArrays(0, [1, 3], [2]))