Link Search Menu Expand Document

4. Median of Two Sorted Arrays

Solution Code

Java

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        double median = 0.0;
        int m = nums1.length;
        int n = nums2.length;
        int combinedArr[] = new int[m + n];
        int i = 0, j = 0, k = 0; 
        
        while (i < m && j < n) { 
            if (nums1[i] <= nums2[j]) { 
                combinedArr[k] = nums1[i++]; 
            } else { 
                combinedArr[k] = nums2[j++]; 
            }
            
            k++; 
        }
        
        while (i < m) { 
            combinedArr[k] = nums1[i]; 
            i++; 
            k++; 
        } 
        
        while (j < n) { 
            combinedArr[k] = nums2[j]; 
            j++; 
            k++; 
        }
        
        if (combinedArr.length % 2 == 0) {
            median = (combinedArr[(combinedArr.length / 2) - 1] + combinedArr[combinedArr.length / 2]) / 2.0;
        } else {
            median = (combinedArr[combinedArr.length / 2]) / 1.0;
        }
        
        return median;   
    }
}

Python

# This solution is for python3, may or may not work for other versions

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

        if len(nums3) % 2 != 0:
            return nums3[(len(nums3) // 2)]
        else:
            return (
                nums3[(len(nums3) // 2) - 1] +
                nums3[(len(nums3) // 2)]
            ) / 2

© 2023. All rights reserved.

pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy