hacking or software engineering skills/leetcode

4. Median of Two Sorted Arrays

study&grow 2019. 5. 9. 11:19
728x90

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

You may assume nums1 and nums2 cannot be both empty.

Example 1:

nums1 = [1, 3] nums2 = [2] The median is 2.0

Example 2:

nums1 = [1, 2] nums2 = [3, 4] The median is (2 + 3)/2 = 2.5

 

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        concatenate = nums1 + nums2
        concatenate.sort()
        if len(concatenate) % 2 != 0:
            return concatenate[len(concatenate)/2]
        else:
            return (concatenate[len(concatenate)/2-1] + concatenate[len(concatenate)/2])/2.