[Leetcode] 4. Median of Two Sorted Arrays

Median of Two Sorted Arrays - LeetCode

Can you solve this real interview question? Median of Two Sorted Arrays - Given two sorted arrays nums1 and nums2 of size m and n respectively, return the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)). Example 1: Input: nums1 = [1,3], nums2 = [2] Output: 2.00000 Explanation: merged array = [1,2,3] and median is 2. Example 2: Input: nums1 = [1,2], nums2 = [3,4] Output: 2.50000 Explanation: merged array = [1,2,3,4] and median is (2 + 3) / 2 = 2.5. Constraints: * nums1.length == m * nums2.length == n * 0 <= m <= 1000 * 0 <= n <= 1000 * 1 <= m + n <= 2000 * -106 <= nums1[i], nums2[i] <= 106

https://leetcode.com/problems/median-of-two-sorted-arrays/description/

🖥️ 시작하며

간단한 배열 관련 문제입니다.

이 문제를 해결하기 위해 먼저 두 배열을 하나의 배열로 병합하고, 병합된 배열을 정렬한 다음 중앙값을 찾는 방법을 사용했습니다.

배열 병합:

nums1 과 nums2 를 더하여 하나의 배열로 병합합니다.
sorted() 함수를 사용하여 병합된 배열을 정렬합니다.

중앙값 찾기:

병합된 배열의 길이를 변수 N에 저장합니다.
배열의 중간 인덱스를 변수 mid에 저장합니다.
배열의 길이가 홀수인 경우 중간 인덱스에 해당하는 값을 반환합니다.
배열의 길이가 짝수인 경우 중간 두 값의 평균을 반환합니다.

⚙️ Python


from typing import List


class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        merged_arr = sorted(nums1 + nums2)

        N = len(merged_arr)
        mid = N // 2

        if N % 2 == 1:
            return merged_arr[mid]
        else:
            return (merged_arr[mid - 1] + merged_arr[mid]) / 2

[Leetcode] 4. Median of Two Sorted Arrays

🖥️ 시작하며

⚙️ Python

댓글