본문 바로가기

hacking or software engineering skills/leetcode

4. Median of Two Sorted Arrays

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.
        

'hacking or software engineering skills > leetcode' 카테고리의 다른 글

6. ZigZag Conversion  (0) 2019.05.09
5. Longest Palindromic Substring  (0) 2019.05.09
3. Longest Substring Without Repeating Characters  (0) 2019.05.09
2. Add Two Numbers  (0) 2019.05.09
1. two Sum  (0) 2019.05.09