Median of two sorted Arrays of different sizes
Given two sorted arrays, a[] and b[], the task is to find the median of these sorted arrays, where N is the number of elements in the first array, and M is the number of elements in the second array.
This is an extension of median of two sorted arrays of equal size problem. Here we handle arrays of unequal size also.
Examples:
Input: a[] = {-5, 3, 6, 12, 15}, b[] = {-12, -10, -6, -3, 4, 10}
Output: The median is 3.
Explanation: The merged array is: ar3[] = {-12, -10, -6, -5 , -3, 3, 4, 6, 10, 12, 15}.
So the median of the merged array is 3Input: a[] = {2, 3, 5, 8}, b[] = {10, 12, 14, 16, 18, 20}
Output: The median is 11.
Explanation : The merged array is: ar3[] = {2, 3, 5, 8, 10, 12, 14, 16, 18, 20}
If the number of the elements are even. So there are two middle elements.
Take the average between the two: (10 + 12) / 2 = 11.
The idea is to merge them into third array and there are two cases:
- Case 1: If the length of the third array is odd, then the median is at (length)/2th index in the array obtained after merging both the arrays.
- Case 2: If the length of the third array is even, then the median will be the average of elements at index ((length)/2 ) and ((length)/2 – 1) in the array obtained after merging both arrays.
Illustration:
arr1[] = { -5, 3, 6, 12, 15 } , arr2[] = { -12, -10, -6, -3, 4, 10 }
- After merging them in a third array : arr3[] = { -5, 3, 6, 12, 15, -12, -10, -6, -3, 4, 10}
- Sort arr3[ ] = { -12, -10, -6, -5, -3, 3, 4, 6, 10, 12, 15 }
- As the length of arr3 is odd, so the median is 3
Follow the steps below to solve the problem:
- Merge the two given arrays into one array.
- Then sort the third(merged) array
- If the length of the third array is even then:
- Divide the length of array by 2. Return (arr[value] + arr[value – 1] / 2).
- If the length of the third array is odd then:
- Divide the length of the array by 2 and round that value and return the arr[value]
Below is the implementation of the above approach:
C++
// C++ program for the above approach #include <bits/stdc++.h> using namespace std; int Solution( int arr[], int n) { // If length of array is even if (n % 2 == 0) { int z = n / 2; int e = arr[z]; int q = arr[z - 1]; int ans = (e + q) / 2; return ans; } // If length if array is odd else { int z = round(n / 2); return arr[z]; } } // Driver Code int main() { // TODO Auto-generated method stub int arr1[] = { -5, 3, 6, 12, 15 }; int arr2[] = { -12, -10, -6, -3, 4, 10 }; int i = sizeof (arr1) / sizeof (arr1[0]); int j = sizeof (arr2) / sizeof (arr2[0]); int arr3[i+j]; int l = i+j; // Merge two array into one array for ( int k=0;k<i;k++) { arr3[k]=arr1[k]; } int a=0; for ( int k=i;k<l;k++) { arr3[k]=arr2[a++]; } // Sort the merged array sort(arr3,arr3+l); // calling the method cout<< "Median = " << Solution(arr3, l); } // This code is contributed by SoumikMondal |
Java
// Java program for the above approach import java.io.*; import java.util.Arrays; public class GFG { public static int Solution( int [] arr) { int n = arr.length; // If length of array is even if (n % 2 == 0 ) { int z = n / 2 ; int e = arr[z]; int q = arr[z - 1 ]; int ans = (e + q) / 2 ; return ans; } // If length if array is odd else { int z = Math.round(n / 2 ); return arr[z]; } } // Driver Code public static void main(String[] args) { // TODO Auto-generated method stub int [] arr1 = { - 5 , 3 , 6 , 12 , 15 }; int [] arr2 = { - 12 , - 10 , - 6 , - 3 , 4 , 10 }; int i = arr1.length; int j = arr2.length; int [] arr3 = new int [i + j]; // Merge two array into one array System.arraycopy(arr1, 0 , arr3, 0 , i); System.arraycopy(arr2, 0 , arr3, i, j); // Sort the merged array Arrays.sort(arr3); // calling the method System.out.print( "Median = " + Solution(arr3)); } } // This code is contributed by Manas Tole |
Python3
# Python3 program for the above approach def Solution(arr): n = len (arr) # If length of array is even if n % 2 = = 0 : z = n / / 2 e = arr[z] q = arr[z - 1 ] ans = (e + q) / 2 return ans # If length of array is odd else : z = n / / 2 ans = arr[z] return ans # Driver code if __name__ = = "__main__" : arr1 = [ - 5 , 3 , 6 , 12 , 15 ] arr2 = [ - 12 , - 10 , - 6 , - 3 , 4 , 10 ] # Concatenating the two arrays arr3 = arr1 + arr2 # Sorting the resultant array arr3.sort() print ( "Median = " , Solution(arr3)) # This code is contributed by kush11 |
C#
// C# program for the above approach using System; using System.Collections.Generic; public class GFG { public static int Solution( int [] arr) { int n = arr.Length; // If length of array is even if (n % 2 == 0) { int z = n / 2; int e = arr[z]; int q = arr[z - 1]; int ans = (e + q) / 2; return ans; } // If length if array is odd else { int z = n / 2; return arr[z]; } } // Driver Code static public void Main (){ // TODO Auto-generated method stub int [] arr1 = { -5, 3, 6, 12, 15 }; int [] arr2 = { -12, -10, -6, -3, 4, 10 }; // Merge two array into one array var myList = new List< int >(); myList.AddRange(arr1); myList.AddRange(arr2); int [] arr3 = myList.ToArray(); // Sort the merged array Array.Sort(arr3); // calling the method Console.Write( "Median = " + Solution(arr3)); } } // This code is contributed by Shubhamsingh10 |
Javascript
<script> // Javascript program for the above approach function Solution(arr, n) { // If length of array is even if (n % 2 == 0) { var z = n / 2; var e = arr[z]; var q = arr[z - 1]; var ans = (e + q) / 2; return ans; } // If length if array is odd else { var z = Math.floor(n / 2); return arr[z]; } } // Driver Code // TODO Auto-generated method stub var arr1 = [ -5, 3, 6, 12, 15 ]; var arr2 = [ -12, -10, -6, -3, 4, 10 ]; var i = arr1.length; var j = arr2.length; var l = i+j; // Merge two array into one array const arr3 = arr1.concat(arr2); // Sort the merged array arr3.sort( function (a, b) { return a - b; }); // calling the method document.write( "Median = " + Solution(arr3, l)); // This code is contributed by Shubham Singh </script> |
Median = 3
Time Complexity: O((N + M) Log (N + M)), Time required to sort the array of size N + M
Auxiliary Space: O(N + M), Creating a new array of size N+M.
Median of two sorted arrays of different sizes by Merging Arrays efficiently:
The given arrays are sorted, so merge the sorted arrays in an efficient way and keep the count of elements inserted in the output array or printed form. So when the elements in the output array are half the original size of the given array print the element as a median element. There are two cases:
- Case 1: M+N is odd, the median is at (M+N)/2th index in the array obtained after merging both the arrays.
- Case 2: M+N is even, the median will be the average of elements at index ((M+N)/2 – 1) and (M+N)/2 in the array obtained after merging both the arrays
Illustration:
Given two array ar1[ ]= { 900 } and ar2[ ] = { 5, 8, 10, 20 } , n => Size of ar1 = 1 and m => Size of ar2 = 4
- Loop will run from 0 till 2.
- First iteration : { 900 } { 5, 8, 10, 20 } , m1 = 5
- Second iteration : { 900 } { 5, 8, 10, 20 }, m1 = 8
- Third iteration : { 900 } { 5, 8, 10, 20 }, m1 = 10
- As size of ar1 + ar2 = odd , hence we return m1 = 10 as the median
Follow the steps below to solve the problem:
- Given two arrays are sorted. So they can be merged in O(m+n) time. Create a variable count to have a count of elements in the output array.
- If value of (M+N) is odd, then there is only one median else the median is the average of elements at index (M+N)/2 and ((M+N)/2 – 1).
- To merge both arrays, keep two indices i and j initially assigned to 0. Compare the ith index of 1st array and jth index of the second, increase the index of the smallest element and increase the count.
- Store (M+N)/2 and (M+N)/2-1 in two variables.
- Check if the count reached (M+N) / 2. If (M+N) is odd return m1. If even return (m1+m2)/2.
Below is the implementation of the above approach:
C++
// A Simple Merge based O(n) solution to find // median of two sorted arrays #include <bits/stdc++.h> using namespace std; /* This function returns median of ar1[] and ar2[]. Assumption in this function: Both ar1[] and ar2[] are sorted arrays */ int getMedian( int ar1[], int ar2[], int n, int m) { int i = 0; /* Current index of input array ar1[] */ int j = 0; /* Current index of input array ar2[] */ int count; int m1 = -1, m2 = -1; /*loop till (m+n)/2*/ for (count = 0; count <= (m + n) / 2; count++) { // store (n+m)/2-1 in m2 m2 = m1; if (i != n && j != m) { m1 = (ar1[i] > ar2[j]) ? ar2[j++] : ar1[i++]; } else if (i < n) { m1 = ar1[i++]; } // for case when j<m, else { m1 = ar2[j++]; } } // Since there are (n+m) elements, // There are following two cases // if n+m is odd then the middle // index is median i.e. (m+n)/2 // other wise median will be average of elements // at index ((m+n)/2 - 1) and (m+n)/2 // in the array obtained after merging ar1 and ar2 if ((m + n) % 2 == 1) { return m1; } else { return (m1 + m2) / 2; } } /* Driver code */ int main() { int ar1[] = { 900 }; int ar2[] = { 5, 8, 10, 20 }; int n1 = sizeof (ar1) / sizeof (ar1[0]); int n2 = sizeof (ar2) / sizeof (ar2[0]); cout << getMedian(ar1, ar2, n1, n2); } // This is code is contributed by rathbhupendra, modified by // rajesh999 |
Java
// A Simple Merge based O(n) solution // to find median of two sorted arrays import java.io.*; class GFG { // Function to calculate median static int getMedian( int ar1[], int ar2[], int n, int m) { // Current index of input array ar1[] int i = 0 ; // Current index of input array ar2[] int j = 0 ; int count; int m1 = - 1 , m2 = - 1 ; // Since there are (n+m) elements, // There are following two cases // if n+m is odd then the middle // index is median i.e. (m+n)/2 if ((m + n) % 2 == 1 ) { for (count = 0 ; count <= (n + m) / 2 ; count++) { if (i != n && j != m) { m1 = (ar1[i] > ar2[j]) ? ar2[j++] : ar1[i++]; } else if (i < n) { m1 = ar1[i++]; } // for case when j<m, else { m1 = ar2[j++]; } } return m1; } // median will be average of elements // at index ((m+n)/2 - 1) and (m+n)/2 // in the array obtained after merging // ar1 and ar2 else { for (count = 0 ; count <= (n + m) / 2 ; count++) { m2 = m1; if (i != n && j != m) { m1 = (ar1[i] > ar2[j]) ? ar2[j++] : ar1[i++]; } else if (i < n) { m1 = ar1[i++]; } // for case when j<m, else { m1 = ar2[j++]; } } return (m1 + m2) / 2 ; } } // Driver code public static void main(String[] args) { int ar1[] = { 900 }; int ar2[] = { 5 , 8 , 10 , 20 }; int n1 = ar1.length; int n2 = ar2.length; System.out.println(getMedian(ar1, ar2, n1, n2)); } } // This code is contributed by Yash Singhal |
Python3
# A Simple Merge based O(n) solution to find # median of two sorted arrays """ This function returns median of ar1[] and ar2[]. Assumption in this function: Both ar1[] and ar2[] are sorted arrays """ def getMedian(ar1, ar2, n, m): i = 0 # Current index of input array ar1[] j = 0 # Current index of input array ar2[] m1, m2 = - 1 , - 1 for count in range (((n + m) / / 2 ) + 1 ): if (i ! = n and j ! = m): if ar1[i] > ar2[j]: m1 = ar2[j] j + = 1 else : m1 = ar1[i] i + = 1 elif (i < n): m1 = ar1[i] i + = 1 # for case when j<m, else : m1 = ar2[j] j + = 1 # return m1 if it's length odd else return (m1+m2)//2 return m1 if (n + m) % 2 = = 1 else (m1 + m2) / / 2 # Driver code ar1 = [ 900 ] ar2 = [ 5 , 8 , 10 , 20 ] n1 = len (ar1) n2 = len (ar2) print (getMedian(ar1, ar2, n1, n2)) # This code is contributed by divyesh072019 |
C#
// A Simple Merge based O(n) solution // to find median of two sorted arrays using System; class GFG { // Function to calculate median static int getMedian( int [] ar1, int [] ar2, int n, int m) { // Current index of input array ar1[] int i = 0; // Current index of input array ar2[] int j = 0; int count; int m1 = -1, m2 = -1; // Since there are (n+m) elements, // There are following two cases // if n+m is odd then the middle // index is median i.e. (m+n)/2 if ((m + n) % 2 == 1) { for (count = 0; count <= (n + m) / 2; count++) { if (i != n && j != m) { m1 = (ar1[i] > ar2[j]) ? ar2[j++] : ar1[i++]; } else if (i < n) { m1 = ar1[i++]; } // for case when j<m, else { m1 = ar2[j++]; } } return m1; } // median will be average of elements // at index ((m+n)/2 - 1) and (m+n)/2 // in the array obtained after merging // ar1 and ar2 else { for (count = 0; count <= (n + m) / 2; count++) { m2 = m1; if (i != n && j != m) { m1 = (ar1[i] > ar2[j]) ? ar2[j++] : ar1[i++]; } else if (i < n) { m1 = ar1[i++]; } // for case when j<m, else { m1 = ar2[j++]; } } return (m1 + m2) / 2; } } // Driver code public static void Main(String[] args) { int [] ar1 = { 900 }; int [] ar2 = { 5, 8, 10, 20 }; int n1 = ar1.Length; int n2 = ar2.Length; Console.WriteLine(getMedian(ar1, ar2, n1, n2)); } } // This code is contributed by Princi Singh |
Javascript
<script> // A Simple Merge based O(n) solution to find // median of two sorted arrays // This function returns median of ar1[] and ar2[]. // Assumption in this function: // Both ar1[] and ar2[] are sorted arrays function getMedian(ar1, ar2, n, m) { // Current index of input array ar1[] let i = 0; // Current index of input array ar2[] let j = 0; let count; let m1 = -1, m2 = -1; // Since there are (n+m) elements, // There are following two cases // if n+m is odd then the middle // index is median i.e. (m+n)/2 if ((m + n) % 2 == 1) { for (count = 0; count <= (n + m) / 2; count++) { if (i != n && j != m) { m1 = (ar1[i] > ar2[j]) ? ar2[j++] : ar1[i++]; } else if (i < n) { m1 = ar1[i++]; } // For case when j<m, else { m1 = ar2[j++]; } } return m1; } // Median will be average of elements // at index ((m+n)/2 - 1) and (m+n)/2 // in the array obtained after merging // ar1 and ar2 else { for (count = 0; count <= (n + m) / 2; count++) { m2 = m1; if (i != n && j != m) { m1 = (ar1[i] > ar2[j]) ? ar2[j++] : ar1[i++]; } else if (i < n) { m1 = ar1[i++]; } // For case when j<m, else { m1 = ar2[j++]; } } return (m1 + m2) / 2; } } // Driver code let ar1 = [900]; let ar2 = [5, 8, 10, 20]; let n1 = ar1.length; let n2 = ar2.length; document.write(getMedian(ar1, ar2, n1, n2)); // This code is contributed by divyeshrabadiya07 </script> |
10
Time Complexity: O(M + N). To merge both arrays O(M+N) time is needed.
Auxiliary Space: O(1). No extra space is required.
Median of two sorted arrays of different sizes using Recursion:
The idea is simple, calculate the median of both arrays and discard one-half of each array. This approach takes into consideration the size of the arrays. The smaller-sized array is considered the first array in the parameter.
Illustration :
Let’s take an example to understand this
Input :arr[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10},
brr[] = { 11, 12, 13, 14, 15, 16, 17, 18, 19 }Recursive call 1:
smaller array[] = 1 2 3 4 5 6 7 8 9 10, mid = 5
larger array[] = 11 12 13 14 15 16 17 18 19 , mid = 155 < 15
Discard first half of the first array and second half of the second arrayRecursive call 2:
smaller array[] = 11 12 13 14 15, mid = 13
larger array[] = 5 6 7 8 9 10, mid = 77 < 13
Discard first half of the second array and second half of the first arrayRecursive call 3:
smaller array[] = 11 12 13 , mid = 12
larger array[] = 7 8 9 10 , mid = 88 < 12
Discard first half of the second array and second half of the first arrayRecursive call 4:
smaller array[] = 11 12
larger array[] = 8 9 10Size of the smaller array is 2 and the size of the larger array is odd
so, the median will be the median of max( 11, 8), 9, min( 10, 12)
that is 9, 10, 11, so the median is 10.Output:10.000000
Some corner cases:
- If the size of the smaller array is 0. Return the median of a larger array.
- if the size of the smaller array is 1.
- The size of the larger array is also 1. Return the median of two elements.
- If the size of the larger array is odd. Then after adding the element from the 2nd array, it will be even so the median will be an average of two mid elements.
- So the element from the smaller array will affect the median if and only if it lies between (M/2 – 1)th and (M/2 + 1)th element of the larger array.
- So, find the median in between the four elements, the element of the smaller array and (M/2)th, (M/2 – 1)th, (M/2 + 1)th element of a larger array
- Similarly, if size is even, then check for the median of three elements, the element of the smaller array and (M/2)th, (M/2 – 1)th element of a larger array
- If the size of the smaller array is 2
- If the larger array also has two elements, find the median of four elements.
- If the larger array has an odd number of elements, then the median will be one of the following 3 elements
- The middle element of the larger array
- Max of the second element of smaller array and element just before the middle, i.e M/2-1th element in a bigger array
- Min of the first element of smaller array and element
just after the middle in the bigger array, i.e M/2 + 1th element in the bigger array
- If the larger array has an even number of elements, then the median will be one of the following 4 elements
- The middle two elements of the larger array
- Max of the first element of smaller array and element just before the first middle element in the bigger array, i.e M/2 – 2nd element
- Min of the second element of smaller array and element just after the second middle in the bigger array, M/2 + 1th element
Follow the steps below to solve the problem:
- Create a recursive function that takes two arrays and the sizes of both arrays.
- Take care of the base cases for the size of arrays less than 2. (previously discussed in Approach).Note: The first array is always the smaller array.
- Find the middle elements of both arrays. i.e element at (n – 1)/2 and (m – 1)/2 of first and second array respectively. Compare both elements.
- If the middle element of the smaller array is less than the middle element of the larger array then the first half of the smaller array is bound to lie strictly in the first half of the merged array. It can also be stated that there is an element in the first half of the larger array and the second half of the smaller array which is the median. So, reduce the search space to the first half of the larger array and the second half of the smaller array.
- Similarly, If the middle element of the smaller array is greater than the middle element of the larger array then reduce the search space to the first half of the smaller array and the second half of the larger array.
Below is the implementation of the above approach:
C++
// A C++ program to find median of two sorted arrays of // unequal sizes #include <bits/stdc++.h> using namespace std; // A utility function to find median of two integers float MO2( int a, int b) { return (a + b) / 2.0; } // A utility function to find median of three integers float MO3( int a, int b, int c) { return a + b + c - max(a, max(b, c)) - min(a, min(b, c)); } // A utility function to find a median of four integers float MO4( int a, int b, int c, int d) { int Max = max(a, max(b, max(c, d))); int Min = min(a, min(b, min(c, d))); return (a + b + c + d - Max - Min) / 2.0; } // Utility function to find median of single array float medianSingle( int arr[], int n) { if (n == 0) return -1; if (n % 2 == 0) return ( double )(arr[n / 2] + arr[n / 2 - 1]) / 2; return arr[n / 2]; } // This function assumes that N is smaller than or equal to // M This function returns -1 if both arrays are empty float findMedianUtil( int A[], int N, int B[], int M) { // If smaller array is empty, return median from second // array if (N == 0) return medianSingle(B, M); // If the smaller array has only one element if (N == 1) { // Case 1: If the larger array also has one element, // simply call MO2() if (M == 1) return MO2(A[0], B[0]); // Case 2: If the larger array has odd number of // elements, then consider the middle 3 elements of // larger array and the only element of smaller // array. Take few examples like following A = {9}, // B[] = {5, 8, 10, 20, 30} and A[] = {1}, B[] = {5, // 8, 10, 20, 30} if (M & 1) return MO2(B[M / 2], MO3(A[0], B[M / 2 - 1], B[M / 2 + 1])); // Case 3: If the larger array has even number of // element, then median will be one of the following // 3 elements // ... The middle two elements of larger array // ... The only element of smaller array return MO3(B[M / 2], B[M / 2 - 1], A[0]); } // If the smaller array has two elements else if (N == 2) { // Case 4: If the larger array also has two // elements, simply call MO4() if (M == 2) return MO4(A[0], A[1], B[0], B[1]); // Case 5: If the larger array has odd number of // elements, then median will be one of the // following 3 elements // 1. Middle element of larger array // 2. Max of first element of smaller array and // element // just before the middle in bigger array // 3. Min of second element of smaller array and // element // just after the middle in bigger array if (M & 1) return MO3(B[M / 2], max(A[0], B[M / 2 - 1]), min(A[1], B[M / 2 + 1])); // Case 6: If the larger array has even number of // elements, then median will be one of the // following 4 elements 1) & 2) The middle two // elements of larger array 3) Max of first element // of smaller array and element // just before the first middle element in bigger // array // 4. Min of second element of smaller array and // element // just after the second middle in bigger array return MO4(B[M / 2], B[M / 2 - 1], max(A[0], B[M / 2 - 2]), min(A[1], B[M / 2 + 1])); } int idxA = (N - 1) / 2; int idxB = (M - 1) / 2; /* if A[idxA] <= B[idxB], then median must exist in A[idxA....] and B[....idxB] */ if (A[idxA] <= B[idxB]) return findMedianUtil(A + idxA, N / 2 + 1, B, M - idxA); /* if A[idxA] > B[idxB], then median must exist in A[...idxA] and B[idxB....] */ return findMedianUtil(A, N / 2 + 1, B + idxA, M - idxA); } // A wrapper function around findMedianUtil(). This function // makes sure that smaller array is passed as first argument // to findMedianUtil float findMedian( int A[], int N, int B[], int M) { if (N > M) return findMedianUtil(B, M, A, N); return findMedianUtil(A, N, B, M); } // Driver program to test above functions int main() { int A[] = { 900 }; int B[] = { 5, 8, 10, 20 }; int N = sizeof (A) / sizeof (A[0]); int M = sizeof (B) / sizeof (B[0]); printf ( "%f" , findMedian(A, N, B, M)); return 0; } |
Java
// A Java program to find median of two sorted arrays of // unequal sizes import java.util.*; class GFG { // A utility function to find median of two integers static float MO2( int a, int b) { return ( float )((a + b) / 2.0 ); } // A utility function to find median of three integers static float MO3( int a, int b, int c) { return a + b + c - Math.max(a, Math.max(b, c)) - Math.min(a, Math.min(b, c)); } // A utility function to find a median of four integers static float MO4( int a, int b, int c, int d) { int Max = Math.max(a, Math.max(b, Math.max(c, d))); int Min = Math.min(a, Math.min(b, Math.min(c, d))); return ( float )((a + b + c + d - Max - Min) / 2.0 ); } // Utility function to find median of single array static float medianSingle( int arr[], int n) { if (n == 0 ) return - 1 ; if (n % 2 == 0 ) return ( float )(( double )(arr[n / 2 ] + arr[n / 2 - 1 ]) / 2 ); return arr[n / 2 ]; } // This function assumes that N is smaller than or equal // to M This function returns -1 if both arrays are // empty static float findMedianUtil( int A[], int N, int B[], int M) { // If smaller array is empty, return median from // second array if (N == 0 ) return medianSingle(B, M); // If the smaller array has only one element if (N == 1 ) { // Case 1: If the larger array also has one // element, simply call MO2() if (M == 1 ) return MO2(A[ 0 ], B[ 0 ]); // Case 2: If the larger array has odd number of // elements, then consider the middle 3 elements // of larger array and the only element of // smaller array. Take few examples like // following A = {9}, B[] = {5, 8, 10, 20, 30} // and A[] = {1}, B[] = {5, 8, 10, 20, 30} if (M % 2 == 1 ) return MO2(B[M / 2 ], ( int )MO3(A[ 0 ], B[M / 2 - 1 ], B[M / 2 + 1 ])); // Case 3: If the larger array has even number // of element, then median will be one of the // following 3 elements // ... The middle two elements of larger array // ... The only element of smaller array return MO3(B[M / 2 ], B[M / 2 - 1 ], A[ 0 ]); } // If the smaller array has two elements else if (N == 2 ) { // Case 4: If the larger array also has two // elements, simply call MO4() if (M == 2 ) return MO4(A[ 0 ], A[ 1 ], B[ 0 ], B[ 1 ]); // Case 5: If the larger array has odd number of // elements, then median will be one of the // following 3 elements // 1. Middle element of larger array // 2. Max of first element of smaller array and // element just before the middle in bigger // array // 3. Min of second element of smaller array and // element just after the middle in bigger array if (M % 2 == 1 ) return MO3(B[M / 2 ], Math.max(A[ 0 ], B[M / 2 - 1 ]), Math.min(A[ 1 ], B[M / 2 + 1 ])); // Case 6: If the larger array has even number // of elements, then median will be one of the // following 4 elements 1) & 2) The middle two // elements of larger array 3) Max of first // element of smaller array and element just // before the first middle element in bigger // array // 4. Min of second element of smaller array and // element just after the second middle in // bigger array return MO4(B[M / 2 ], B[M / 2 - 1 ], Math.max(A[ 0 ], B[M / 2 - 2 ]), Math.min(A[ 1 ], B[M / 2 + 1 ])); } int idxA = (N - 1 ) / 2 ; int idxB = (M - 1 ) / 2 ; /* * if A[idxA] <= B[idxB], then median must exist in A[idxA....] and B[....idxB] */ if (A[idxA] <= B[idxB]) return findMedianUtil( Arrays.copyOfRange(A, idxA, A.length), N / 2 + 1 , B, M - idxA); /* * if A[idxA] > B[idxB], then median must exist in A[...idxA] and B[idxB....] */ return findMedianUtil( A, N / 2 + 1 , Arrays.copyOfRange(B, idxB, B.length), M - idxA); } // A wrapper function around findMedianUtil(). This // function makes sure that smaller array is passed as // first argument to findMedianUtil static float findMedian( int A[], int N, int B[], int M) { if (N > M) return findMedianUtil(B, M, A, N); return findMedianUtil(A, N, B, M); } // Driver program to test above functions public static void main(String[] args) { int A[] = { 900 }; int B[] = { 5 , 8 , 10 , 20 }; int N = A.length; int M = B.length; System.out.printf( "%f" , findMedian(A, N, B, M)); } } // This code is contributed by Princi Singh. |
Python3
# A Python3 program to find median of two sorted arrays of # unequal sizes # A utility function to find median of two integers def MO2(a, b): return (a + b) / 2 # A utility function to find median of three integers def MO3(a, b, c): return a + b + c - max (a, max (b, c)) - min (a, min (b, c)) # A utility function to find a median of four integers def MO4(a, b, c, d): Max = max (a, max (b, max (c, d))) Min = min (a, min (b, min (c, d))) return (a + b + c + d - Max - Min ) / 2 # Utility function to find median of single array def medianSingle(arr, n): if (n = = 0 ): return - 1 if (n % 2 = = 0 ): return (arr[n / 2 ] + arr[n / 2 - 1 ]) / 2 return arr[n / 2 ] # This function assumes that N is smaller than or equal to M # This function returns -1 if both arrays are empty def findMedianUtil(A, N, B, M): # If smaller array is empty, return median from second array if (N = = 0 ): return medianSingle(B, M) # If the smaller array has only one element if (N = = 1 ): # Case 1: If the larger array also has one element, # simply call MO2() if (M = = 1 ): return MO2(A[ 0 ], B[ 0 ]) # Case 2: If the larger array has odd number of elements, # then consider the middle 3 elements of larger array and # the only element of smaller array. Take few examples # like following # A = {9}, B[] = {5, 8, 10, 20, 30} and # A[] = {1}, B[] = {5, 8, 10, 20, 30} if (M & 1 ! = 0 ): return MO2(B[M / 2 ], MO3(A[ 0 ], B[M / 2 - 1 ], B[M / 2 + 1 ])) # Case 3: If the larger array has even number of element, # then median will be one of the following 3 elements # ... The middle two elements of larger array # ... The only element of smaller array return MO3(B[M / / 2 ], B[M / / 2 - 1 ], A[ 0 ]) # If the smaller array has two elements elif (N = = 2 ): # Case 4: If the larger array also has two elements, # simply call MO4() if (M = = 2 ): return MO4(A[ 0 ], A[ 1 ], B[ 0 ], B[ 1 ]) # Case 5: If the larger array has odd number of elements, # then median will be one of the following 3 elements # 1. Middle element of larger array # 2. Max of first element of smaller array and element # just before the middle in bigger array # 3. Min of second element of smaller array and element # just after the middle in bigger array if (M & 1 ! = 0 ): return MO3(B[M / 2 ], max (A[ 0 ], B[M / 2 - 1 ]), min (A[ 1 ], B[M / 2 + 1 ])) # Case 6: If the larger array has even number of elements, # then median will be one of the following 4 elements # 1) & 2) The middle two elements of larger array # 3) Max of first element of smaller array and element # just before the first middle element in bigger array # 4. Min of second element of smaller array and element # just after the second middle in bigger array return MO4(B[M / 2 ], B[M / 2 - 1 ], max (A[ 0 ], B[M / 2 - 2 ]), min (A[ 1 ], B[M / 2 + 1 ])) idxA = (N - 1 ) / 2 idxB = (M - 1 ) / 2 ''' if A[idxA] <= B[idxB], then median must exist in A[idxA....] and B[....idxB] ''' if (A[idxA] < = B[idxB]): return findMedianUtil(A + idxA, N / 2 + 1 , B, M - idxA) ''' if A[idxA] > B[idxB], then median must exist in A[...idxA] and B[idxB....] ''' return findMedianUtil(A, N / 2 + 1 , B + idxA, M - idxA) # A wrapper function around findMedianUtil(). This function # makes sure that smaller array is passed as first argument # to findMedianUtil def findMedian(A, N, B, M): if (N > M): return findMedianUtil(B, M, A, N) return findMedianUtil(A, N, B, M) # Driver code A = [ 900 ] B = [ 5 , 8 , 10 , 20 ] N = len (A) M = len (B) print (findMedian(A, N, B, M)) # This code is contributed by divyesh072019 |
C#
// A C# program to find median of two sorted arrays of // unequal sizes using System; class GFG { // A utility function to find median of two integers static float MO2( int a, int b) { return ( float )((a + b) / 2.0); } // A utility function to find median of three integers static float MO3( int a, int b, int c) { return a + b + c - Math.Max(a, Math.Max(b, c)) - Math.Min(a, Math.Min(b, c)); } // A utility function to find a median of four integers static float MO4( int a, int b, int c, int d) { int Max = Math.Max(a, Math.Max(b, Math.Max(c, d))); int Min = Math.Min(a, Math.Min(b, Math.Min(c, d))); return ( float )((a + b + c + d - Max - Min) / 2.0); } // Utility function to find median of single array static float medianSingle( int [] arr, int n) { if (n == 0) return -1; if (n % 2 == 0) return ( float )(( double )(arr[n / 2] + arr[n / 2 - 1]) / 2); return arr[n / 2]; } static int [] copyOfRange( int [] src, int start, int end) { int len = end - start; int [] dest = new int [len]; Array.Copy(src, start, dest, 0, len); return dest; } // This function assumes that N is smaller than or equal // to M This function returns -1 if both arrays are // empty static float findMedianUtil( int [] A, int N, int [] B, int M) { // If smaller array is empty, // return median from second array if (N == 0) return medianSingle(B, M); // If the smaller array has only one element if (N == 1) { // Case 1: If the larger array also has one // element, simply call MO2() if (M == 1) return MO2(A[0], B[0]); // Case 2: If the larger array has odd number of // elements, then consider the middle 3 elements // of larger array and the only element of // smaller array. Take few examples like // following A = {9}, B[] = {5, 8, 10, 20, 30} // and A[] = {1}, B[] = {5, 8, 10, 20, 30} if (M % 2 == 1) return MO2(B[M / 2], ( int )MO3(A[0], B[M / 2 - 1], B[M / 2 + 1])); // Case 3: If the larger array has even number // of element, then median will be one of the // following 3 elements // ... The middle two elements of larger array // ... The only element of smaller array return MO3(B[M / 2], B[M / 2 - 1], A[0]); } // If the smaller array has two elements else if (N == 2) { // Case 4: If the larger array also has two // elements, simply call MO4() if (M == 2) return MO4(A[0], A[1], B[0], B[1]); // Case 5: If the larger array has odd number of // elements, then median will be one of the // following 3 elements // 1. Middle element of larger array // 2. Max of first element of smaller array and // element just before the middle in bigger // array // 3. Min of second element of smaller array and // element just after the middle in bigger array if (M % 2 == 1) return MO3(B[M / 2], Math.Max(A[0], B[M / 2 - 1]), Math.Min(A[1], B[M / 2 + 1])); // Case 6: If the larger array has even number // of elements, then median will be one of the // following 4 elements 1) & 2) The middle two // elements of larger array 3) Max of first // element of smaller array and element just // before the first middle element in bigger // array // 4. Min of second element of smaller array and // element just after the second middle in // bigger array return MO4(B[M / 2], B[M / 2 - 1], Math.Max(A[0], B[M / 2 - 2]), Math.Min(A[1], B[M / 2 + 1])); } int idxA = (N - 1) / 2; int idxB = (M - 1) / 2; /* * if A[idxA] <= B[idxB], then median must exist in A[idxA....] and B[....idxB] */ if (A[idxA] <= B[idxB]) return findMedianUtil( copyOfRange(A, idxA, A.Length), N / 2 + 1, B, M - idxA); /* * if A[idxA] > B[idxB], then median must exist in A[...idxA] and B[idxB....] */ return findMedianUtil( A, N / 2 + 1, copyOfRange(B, idxB, B.Length), M - idxA); } // A wrapper function around findMedianUtil(). This // function makes sure that smaller array is passed as // first argument to findMedianUtil static float findMedian( int [] A, int N, int [] B, int M) { if (N > M) return findMedianUtil(B, M, A, N); return findMedianUtil(A, N, B, M); } // Driver code static void Main() { int [] A = { 900 }; int [] B = { 5, 8, 10, 20 }; int N = A.Length; int M = B.Length; Console.WriteLine(findMedian(A, N, B, M)); } } // This code is contributed by divyeshrabadiya07 |
PHP
<?php // A PHP program to find median // of two sorted arrays of // unequal sizes // A utility function to // find median of two integers function MO2( $a , $b ) { return ( $a + $b ) / 2.0; } // A utility function to // find median of three integers function MO3( $a , $b , $c ) { return $a + $b + $c - max( $a , max( $b , $c )) - min( $a , min( $b , $c )); } // A utility function to find // median of four integers function MO4( $a , $b , $c , $d ) { $Max = max( $a , max( $b , max( $c , $d ))); $Min = min( $a , min( $b , min( $c , $d ))); return ( $a + $b + $c + $d - $Max - $Min ) / 2.0; } // Utility function to // find median of single array function medianSingle( $arr , $n ) { if ( $n == 0) return -1; if ( $n % 2 == 0) return ( $arr [ $n / 2] + $arr [ $n / 2 - 1]) / 2; return $arr [ $n / 2]; } // This function assumes that N // is smaller than or equal to M // This function returns -1 if // both arrays are empty function findMedianUtil(& $A , $N , & $B , $M ) { // If smaller array is empty, // return median from second array if ( $N == 0) return medianSingle( $B , $M ); // If the smaller array // has only one element if ( $N == 1) { // Case 1: If the larger // array also has one // element, simply call MO2() if ( $M == 1) return MO2( $A [0], $B [0]); // Case 2: If the larger array // has odd number of elements, // then consider the middle 3 // elements of larger array and // the only element of smaller // array. Take few examples // like following // $A = array(9), // $B = array(5, 8, 10, 20, 30) // and $A = array(1), // $B = array(5, 8, 10, 20, 30) if ( $M & 1) return MO2( $B [ $M / 2], $MO3 ( $A [0], $B [ $M / 2 - 1], $B [ $M / 2 + 1])); // Case 3: If the larger array // has even number of element, // then median will be one of // the following 3 elements // ... The middle two elements // of larger array // ... The only element of // smaller array return MO3( $B [ $M / 2], $B [ $M / 2 - 1], $A [0]); } // If the smaller array // has two elements else if ( $N == 2) { // Case 4: If the larger // array also has two elements, // simply call MO4() if ( $M == 2) return MO4( $A [0], $A [1], $B [0], $B [1]); // Case 5: If the larger array // has odd number of elements, // then median will be one of // the following 3 elements // 1. Middle element of // larger array // 2. Max of first element of // smaller array and element // just before the middle // in bigger array // 3. Min of second element // of smaller array and element // just after the middle // in bigger array if ( $M & 1) return MO3 ( $B [ $M / 2], max( $A [0], $B [ $M / 2 - 1]), min( $A [1], $B [ $M / 2 + 1])); // Case 6: If the larger array // has even number of elements, // then median will be one of // the following 4 elements // 1) & 2) The middle two // elements of larger array // 3) Max of first element of // smaller array and element // just before the first middle // element in bigger array // 4. Min of second element of // smaller array and element // just after the second // middle in bigger array return MO4 ( $B [ $M / 2], $B [ $M / 2 - 1], max( $A [0], $B [ $M / 2 - 2]), min( $A [1], $B [ $M / 2 + 1])); } $idxA = ( $N - 1 ) / 2; $idxB = ( $M - 1 ) / 2; /* if $A[$idxA] <= $B[$idxB], then median must exist in $A[$idxA....] and $B[....$idxB] */ if ( $A [ $idxA ] <= $B [ $idxB ] ) return findMedianUtil( $A + $idxA , $N / 2 + 1, $B , $M - $idxA ); /* if $A[$idxA] > $B[$idxB], then median must exist in $A[...$idxA] and $B[$idxB....] */ return findMedianUtil( $A , $N /2 + 1, $B + $idxA , $M - $idxA ); } // A wrapper function around // findMedianUtil(). This // function makes sure that // smaller array is passed as // first argument to findMedianUtil function findMedian(& $A , $N , & $B , $M ) { if ( $N > $M ) return findMedianUtil( $B , $M , $A , $N ); return findMedianUtil( $A , $N , $B , $M ); } // Driver Code $A = array (900); $B = array (5, 8, 10, 20); $N = sizeof( $A ); $M = sizeof( $B ); echo findMedian( $A , $N , $B , $M ); // This code is contributed // by ChitraNayal ?> |
Javascript
<script> // A Javascript program to find median of two sorted arrays of // unequal sizes // A utility function to find median of two integers function MO2(a,b) { return ((a + b) / 2.0); } // A utility function to find median of three integers function MO3(a, b, c) { return a + b + c - Math.max(a, Math.max(b, c)) - Math.min(a, Math.min(b, c)); } // A utility function to find a median of four integers function MO4(a, b, c, d) { let Max = Math.max(a, Math.max(b, Math.max(c, d))); let Min = Math.min(a, Math.min(b, Math.min(c, d))); return ((a + b + c + d - Max - Min) / 2.0); } // Utility function to find median of single array function medianSingle(arr, n) { if (n == 0) return -1; if (n % 2 == 0) return ( (arr[n / 2] + arr[n / 2 - 1]) / 2); return arr[n / 2]; } // This function assumes that N is smaller than or equal to M // This function returns -1 if both arrays are empty function findMedianUtil(A, N, B, M) { // If smaller array is empty, return median from second array if (N == 0) return medianSingle(B, M); // If the smaller array has only one element if (N == 1) { // Case 1: If the larger array also has one element, // simply call MO2() if (M == 1) return MO2(A[0], B[0]); // Case 2: If the larger array has odd number of elements, // then consider the middle 3 elements of larger array and // the only element of smaller array. Take few examples // like following // A = {9}, B[] = {5, 8, 10, 20, 30} and // A[] = {1}, B[] = {5, 8, 10, 20, 30} if (M % 2 == 1) return MO2(B[M / 2], MO3(A[0], B[M / 2 - 1], B[M / 2 + 1])); // Case 3: If the larger array has even number of element, // then median will be one of the following 3 elements // ... The middle two elements of larger array // ... The only element of smaller array return MO3(B[M / 2], B[M / 2 - 1], A[0]); } // If the smaller array has two elements else if (N == 2) { // Case 4: If the larger array also has two elements, // simply call MO4() if (M == 2) return MO4(A[0], A[1], B[0], B[1]); // Case 5: If the larger array has odd number of elements, // then median will be one of the following 3 elements // 1. Middle element of larger array // 2. Max of first element of smaller array and element // just before the middle in bigger array // 3. Min of second element of smaller array and element // just after the middle in bigger array if (M % 2 == 1) return MO3(B[M / 2], Math.max(A[0], B[M / 2 - 1]), Math.min(A[1], B[M / 2 + 1])); // Case 6: If the larger array has even number of elements, // then median will be one of the following 4 elements // 1) & 2) The middle two elements of larger array // 3) Max of first element of smaller array and element // just before the first middle element in bigger array // 4. Min of second element of smaller array and element // just after the second middle in bigger array return MO4(B[M / 2], B[M / 2 - 1], Math.max(A[0], B[M / 2 - 2]), Math.min(A[1], B[M / 2 + 1])); } let idxA = (N - 1) / 2; let idxB = (M - 1) / 2; /* * if A[idxA] <= B[idxB], then median must exist in A[idxA....] and B[....idxB] */ if (A[idxA] <= B[idxB]) return findMedianUtil(A.slice(idxA, A.length), N / 2 + 1, B, M - idxA); /* * if A[idxA] > B[idxB], then median must exist in A[...idxA] and B[idxB....] */ return findMedianUtil(A, N / 2 + 1, B.slice( idxB, B.length), M - idxA); } // A wrapper function around findMedianUtil(). This function // makes sure that smaller array is passed as first argument // to findMedianUtil function findMedian(A,N,B,M) { if (N > M) return findMedianUtil(B, M, A, N); return findMedianUtil(A, N, B, M); } // Driver program to test above functions let A = [ 900]; let B = [5, 8, 10, 20]; let N = A.length; let M = B.length; document.write(findMedian(A, N, B, M)); // This code is contributed by avanitrachhadiya2155 </script> |
10.000000
Time Complexity: O(min(log M, log N)). In each step, one-half of each array is discarded. So the algorithm takes O(min(log M, log N)) time to reach the median value.
Auxiliary Space: O(1). No extra space is required.
Median of two sorted arrays of different sizes using Binary Search:
The given two arrays are sorted, so we can utilize the ability of Binary Search to divide the array and find the median. Median means the point at which the whole array is divided into two parts. Hence since the two arrays are not merged so to get the median we require merging which is costly. Hence instead of merging, we will use a modified binary search algorithm to efficiently find the median.
Illustration:
A[ ] = { -5, 3, 6, 12, 15 }, n = 5 & B[ ] = { -12, -10, -6, -3, 4, 10} , m = 6
- realmidinmergedarray = 6.
- start = 0 and end = 5 => mid = 2
- leftAsize = 2 and leftBsize = 4
- leftA = 3
- leftB = -3
- rightA = 6
- rightB = 4
- A[ ] = { -5, 3, 6, 12, 15 } & B[ ] = { -12, -10, -6, -3, 4, 10}
- As leftA <= rightB and leftB <= rightA, so the condition holds and 3 is returned as the median
Follow the steps below to solve the problem:
- If we would have merged the two arrays, the median is the point that will divide the sorted merged array into two equal parts. So the actual median point in the merged array would have been (M+N+1)/2;
- We divide A[] and B[] into two parts. We will find the mid value and divide the first array A[] into two parts and simultaneously choose only those elements from left of B[] array such that the sum of the count of elements in the left part of both A[] and B[] will result in the left part of the merged array.
- Now we have 4 variables indicating four values two from array A[] and two from array B[].
- leftA -> Rightmost element in left part of A.
- leftb -> Rightmost element in left part of B
- rightA -> Leftmost element in right part of A
- rightB -> Leftmost element in right part of B
- Hence to confirm that the partition was correct we have to check if leftA<=rightB and leftB<=rightA. This is the case when the sum of two parts of A and B results in the left part of the merged array.
- If the condition fails we have to find another midpoint in A and then left part in B[].
- If we find leftA > rightB. means we have to decrease the size of A’s partition and shift to lesser value in A[].
- So update the right pointer of to mid-1 else we will increase the left pointer to mid+1.
- Repeat the above steps with new partitions till we get the answers.
- If leftA ≤ rightB and leftB ≤ rightA, then we get the correct partition and our answer depends on the total size of the merged array (i.e. M+N). If (M+N) is even we take max(leftA, leftB) and min(rightA, rightB), add them and divide by 2 to get our answer, else we will just return the maximum of leftA and leftB.
Below is the implementation of the above problem:
C++
#include <bits/stdc++.h> using namespace std; // Method to find median double Median(vector< int >& A, vector< int >& B) { int n = A.size(); int m = B.size(); if (n > m) return Median(B, A); // Swapping to make A smaller int start = 0; int end = n; int realmidinmergedarray = (n + m + 1) / 2; while (start <= end) { int mid = (start + end) / 2; int leftAsize = mid; int leftBsize = realmidinmergedarray - mid; int leftA = (leftAsize > 0) ? A[leftAsize - 1] : INT_MIN; // checking overflow of indices int leftB = (leftBsize > 0) ? B[leftBsize - 1] : INT_MIN; int rightA = (leftAsize < n) ? A[leftAsize] : INT_MAX; int rightB = (leftBsize < m) ? B[leftBsize] : INT_MAX; // if correct partition is done if (leftA <= rightB and leftB <= rightA) { if ((m + n) % 2 == 0) return (max(leftA, leftB) + min(rightA, rightB)) / 2.0; return max(leftA, leftB); } else if (leftA > rightB) { end = mid - 1; } else start = mid + 1; } return 0.0; } // Driver code int main() { vector< int > arr1 = { -5, 3, 6, 12, 15 }; vector< int > arr2 = { -12, -10, -6, -3, 4, 10 }; cout << "Median of the two arrays are" << endl; cout << Median(arr1, arr2); return 0; } |
Java
public class GFG { // Method to find median static double Median( int [] A, int [] B) { int n = A.length; int m = B.length; if (n > m) return Median(B, A); // Swapping to make A smaller int start = 0 ; int end = n; int realmidinmergedarray = (n + m + 1 ) / 2 ; while (start <= end) { int mid = (start + end) / 2 ; int leftAsize = mid; int leftBsize = realmidinmergedarray - mid; int leftA = (leftAsize > 0 ) ? A[leftAsize - 1 ] : Integer .MIN_VALUE; // checking overflow // of indices int leftB = (leftBsize > 0 ) ? B[leftBsize - 1 ] : Integer.MIN_VALUE; int rightA = (leftAsize < n) ? A[leftAsize] : Integer.MAX_VALUE; int rightB = (leftBsize < m) ? B[leftBsize] : Integer.MAX_VALUE; // if correct partition is done if (leftA <= rightB && leftB <= rightA) { if ((m + n) % 2 == 0 ) return (Math.max(leftA, leftB) + Math.min(rightA, rightB)) / 2.0 ; return Math.max(leftA, leftB); } else if (leftA > rightB) { end = mid - 1 ; } else start = mid + 1 ; } return 0.0 ; } // Driver code public static void main(String[] args) { int [] arr1 = { - 5 , 3 , 6 , 12 , 15 }; int [] arr2 = { - 12 , - 10 , - 6 , - 3 , 4 , 10 }; System.out.println( "Median of the two arrays are" ); System.out.println(Median(arr1, arr2)); } } // This code is contributed by Hritik |
Python3
class Solution: # Method to find median def Median( self , A, B): # Assumption both A and B cannot be empty n = len (A) m = len (B) if (n > m): return self .Median(B, A) # Swapping to make A smaller start = 0 end = n realmidinmergedarray = (n + m + 1 ) / / 2 while (start < = end): mid = (start + end) / / 2 leftAsize = mid leftBsize = realmidinmergedarray - mid # checking overflow of indices leftA = A[leftAsize - 1 ] if (leftAsize > 0 ) else float ( '-inf' ) leftB = B[leftBsize - 1 ] if (leftBsize > 0 ) else float ( '-inf' ) rightA = A[leftAsize] if (leftAsize < n) else float ( 'inf' ) rightB = B[leftBsize] if (leftBsize < m) else float ( 'inf' ) # if correct partition is done if leftA < = rightB and leftB < = rightA: if ((m + n) % 2 = = 0 ): return ( max (leftA, leftB) + min (rightA, rightB)) / 2.0 return max (leftA, leftB) elif (leftA > rightB): end = mid - 1 else : start = mid + 1 # Driver code ans = Solution() arr1 = [ - 5 , 3 , 6 , 12 , 15 ] arr2 = [ - 12 , - 10 , - 6 , - 3 , 4 , 10 ] print ( "Median of the two arrays is" ) print (ans.Median(arr1, arr2)) # This code is contributed by Arpan |
C#
using System; public class GFG { // Method to find median static double Median( int [] A, int [] B) { int n = A.Length; int m = B.Length; if (n > m) return Median(B, A); // Swapping to make A smaller int start = 0; int end = n; int realmidinmergedarray = (n + m + 1) / 2; while (start <= end) { int mid = (start + end) / 2; int leftAsize = mid; int leftBsize = realmidinmergedarray - mid; int leftA = (leftAsize > 0) ? A[leftAsize - 1] : Int32.MinValue; // checking overflow // of indices int leftB = (leftBsize > 0) ? B[leftBsize - 1] : Int32.MinValue; int rightA = (leftAsize < n) ? A[leftAsize] : Int32.MaxValue; int rightB = (leftBsize < m) ? B[leftBsize] : Int32.MaxValue; // if correct partition is done if (leftA <= rightB && leftB <= rightA) { if ((m + n) % 2 == 0) return (Math.Max(leftA, leftB) + Math.Min(rightA, rightB)) / 2.0; return Math.Max(leftA, leftB); } else if (leftA > rightB) { end = mid - 1; } else start = mid + 1; } return 0.0; } // Driver code public static void Main() { int [] arr1 = { -5, 3, 6, 12, 15 }; int [] arr2 = { -12, -10, -6, -3, 4, 10 }; Console.WriteLine( "Median of the two arrays are" ); Console.WriteLine(Median(arr1, arr2)); } } // This code is contributed by Shubham Singh |
Javascript
<script> // JavaScript Program to implement // the above approach // Method to find median function Median(A, B) { let n = A.length; let m = B.length; if (n > m) return Median(B, A); // Swapping to make A smaller let start = 0; let end = n; let realmidinmergedarray = Math.floor((n + m + 1) / 2); while (start <= end) { let mid = Math.floor((start + end) / 2); let leftAsize = mid; let leftBsize = realmidinmergedarray - mid; let leftA = (leftAsize > 0) ? A[leftAsize - 1] : Number.MIN_VALUE; // checking overflow of indices let leftB = (leftBsize > 0) ? B[leftBsize - 1] : INT_MIN; let rightA = (leftAsize < n) ? A[leftAsize] : INT_MAX; let rightB = (leftBsize < m) ? B[leftBsize] : INT_MAX; // if correct partition is done if (leftA <= rightB && leftB <= rightA) { if ((m + n) % 2 == 0) return Math.floor((Math.max(leftA, leftB) + Math.min(rightA, rightB)) / 2.0); return Math.max(leftA, leftB); } else if (leftA > rightB) { end = mid - 1; } else start = mid + 1; } return 0.0; } // Driver code let arr1 = [-5, 3, 6, 12, 15]; let arr2 = [-12, -10, -6, -3, 4, 10]; document.write( "Median of the two arrays are" + "<br>" ); document.write(Median(arr1, arr2)) // This code is contributed by Potta Lokesh </script> |
Median of the two arrays are 3
Time Complexity: O(min(log M, log N)): Since binary search is being applied on the smaller of the 2 arrays
Auxiliary Space: O(1)
Median of two sorted arrays of different sizes using Priority Queue:
In this Approach we have used Priority Queue (min Heap) to find out the median. The Idea is simple Just push the elements into a single Priority Queue from both arrays . Now we have to find median from priority queue by performing a simple traversal through it upto median.
Illustration:
A[ ]={-2,3,4,5} ,n=4 & B[ ]={-4,-1,7,8,9},m=5
Step 1:
//Adding elements to priority queue(pq) from array A
pq.push(-2)
pq.push(3)
pq.push(4)
pq.push(5)
After adding array A elements to priority queue it will look as pq={-2,3,4,5}
Step 2:
//Adding elements to priority queue(pq) from array B
pq.push(-4)
pq.push(-1)
pq.push(7)
pq.push(8)
pq.push(9)
After adding array B elements to priority queue it will look as pq={-4,-2,-1,3,4,5,7,8,9}
Step 3:
//Now we have to find median from Priority Queue
Now initialize a count=-1
under Loop increment count to 1 at each pop
check ;
if n+m is odd then traverse priority queue upto (n+m)/2 by popping element by element i.e count==(n+m)/2 then display median as pq.top()
if n+m is even then traverse priority queue upto (n+m)/2 && ((n+m)/2)-1 i.e count ==(n+m)/2 and count==((n+m)/2)-1 maintain both top values of priority queue
In this case the median is 4
Below is the implementation of the above problem:
C++
#include <bits/stdc++.h> using namespace std; // Method to find median double Median(vector< int >& A, vector< int >& B) { int i; int n=A.size(); int m=B.size(); //initializing Priority Queue (Min Heap) priority_queue < int , vector< int >, greater< int >> pq; //pushing array A values to priority Queue for (i=0;i<n;i++) pq.push(A[i]); //pushing array B values to priority Queue for (i=0;i<m;i++) pq.push(B[i]); int check=n+m; double count=-1; double mid1,mid2; while (!pq.empty()) { count++; //returning mid value if combined length(n+m) is odd if (check%2!=0&&count==check/2) { double ans=pq.top(); return ans; } //maintaining mid1 value if combined length(n+m) is even //where we need to maintain both mid values in case of even combined length if (check%2==0&&count==(check/2)-1) mid1=pq.top(); //now returning the mid2 value with previous maintained mid1 value by 2 if (check%2==0&&count==check/2) { mid2=pq.top(); double ans=(mid1+mid2)/2; return ans; } pq.pop(); } return 0.00000; } // Driver code int main() { vector< int > arr1 = {-2,3,4,5}; vector< int > arr2 = { -4,-1,7,8,9 }; cout << "Median of the two arrays are" << endl; cout << Median(arr1, arr2); return 0; } |
Java
import java.util.PriorityQueue; import java.util.Scanner; import java.util.Vector; public class Main { // Method to find median public static double Median(Vector<Integer> A, Vector<Integer> B) { int i; int n = A.size(); int m = B.size(); // initializing Priority Queue (Min Heap) PriorityQueue<Integer> pq = new PriorityQueue<Integer>(); // pushing array A values to priority Queue for (i = 0 ; i < n; i++) { pq.add(A.get(i)); } // pushing array B values to priority Queue for (i = 0 ; i < m; i++) { pq.add(B.get(i)); } int check = n + m; double count = - 1 ; double mid1 = - 1 , mid2 = - 1 ; while (!pq.isEmpty()) { count++; // returning mid value if combined length(n+m) // is odd if (check % 2 != 0 && count == check / 2 ) { double ans = pq.peek(); return ans; } // maintaining mid1 value if combined // length(n+m) is even where we need to maintain // both mid values in case of even combined // length if (check % 2 == 0 && count == (check / 2 ) - 1 ) { mid1 = pq.peek(); } // now returning the mid2 value with previous // maintained mid1 value by 2 if (check % 2 == 0 && count == check / 2 ) { mid2 = pq.peek(); double ans = (mid1 + mid2) / 2 ; return ans; } pq.poll(); } return 0.00000 ; } // Driver code public static void main(String[] args) { Vector<Integer> arr1 = new Vector<Integer>(); arr1.add(- 2 ); arr1.add( 3 ); arr1.add( 4 ); arr1.add( 5 ); Vector<Integer> arr2 = new Vector<Integer>(); arr2.add(- 4 ); arr2.add(- 1 ); arr2.add( 7 ); arr2.add( 8 ); arr2.add( 9 ); System.out.println( "Median of the two arrays are" ); System.out.println(Median(arr1, arr2)); } } |
Python3
# Python code for the above approach import heapq def Median(A, B): n, m = len (A), len (B) # initializing Priority Queue (Min Heap) pq = [] # pushing array A values to priority Queue for i in range (n): heapq.heappush(pq, A[i]) # pushing array B values to priority Queue for i in range (m): heapq.heappush(pq, B[i]) check = n + m count = - 1 mid1, mid2 = - 1 , - 1 while pq: count + = 1 # returning mid value if combined length(n+m) is odd if check % 2 ! = 0 and count = = check / / 2 : return pq[ 0 ] # maintaining mid1 value if combined length(n+m) is even # where we need to maintain both mid values in case of # even combined length if check % 2 = = 0 and count = = (check / / 2 ) - 1 : mid1 = pq[ 0 ] # now returning the mid2 value with previous maintained # mid1 value by 2 if check % 2 = = 0 and count = = check / / 2 : mid2 = heapq.heappop(pq) return (mid1 + mid2) / 2 heapq.heappop(pq) return 0.00000 # Driver code arr1 = [ - 2 , 3 , 4 , 5 ] arr2 = [ - 4 , - 1 , 7 , 8 , 9 ] print ( "Median of the two arrays are" ) print (Median(arr1, arr2)) # This code is contributed by karthik |
Javascript
// Method to find median function Median(A, B) { let n = A.length; let m = B.length; //initializing Priority Queue (Min Heap) let pq = new PriorityQueue(); //pushing array A values to priority Queue for (let i = 0; i < n; i++) { pq.push(A[i]); } //pushing array B values to priority Queue for (let i = 0; i < m; i++) { pq.push(B[i]); } let check = n + m; let count = -1; let mid1, mid2; while (!pq.isEmpty()) { count++; //returning mid value if combined length(n+m) is odd if (check % 2 !== 0 && count === Math.floor(check / 2)) { let ans = pq.top(); return ans; } //maintaining mid1 value if combined length(n+m) is even //where we need to maintain both mid values in case of even combined length if (check % 2 === 0 && count === (check / 2) - 1) { mid1 = pq.top(); } //now returning the mid2 value with previous maintained mid1 value by 2 if (check % 2 === 0 && count === check / 2) { mid2 = pq.top(); let ans = (mid1 + mid2) / 2; return ans; } pq.pop(); } return 0.00000; } // Priority Queue (Min Heap) implementation class PriorityQueue { constructor() { this .data = []; } push(value) { this .data.push(value); this .bubbleUp( this .data.length - 1); } pop() { let result = this .data[0]; let end = this .data.pop(); if ( this .data.length > 0) { this .data[0] = end; this .bubbleDown(0); } return result; } top() { return this .data[0]; } isEmpty() { return this .data.length === 0; } bubbleUp(index) { let parent = Math.floor((index + 1) / 2) - 1; if (parent >= 0 && this .data[parent] > this .data[index]) { [ this .data[parent], this .data[index]] = [ this .data[index], this .data[parent]]; this .bubbleUp(parent); } } bubbleDown(index) { let child1 = (index + 1) * 2 - 1; let child2 = (index + 1) * 2; let minIndex = index; if (child1 < this .data.length && this .data[child1] < this .data[minIndex]) { minIndex = child1; } if (child2 < this .data.length && this .data[child2] < this .data[minIndex]) { minIndex = child2; } if (minIndex !== index) { [ this .data[index], this .data[minIndex]] = [ this .data[minIndex], this .data[index]]; this .bubbleDown(minIndex); } } } // Driver code let arr1 = [-2, 3, 4, 5]; let arr2 = [-4, -1, 7, 8, 9]; console.log( "Median of the two arrays are" ); console.log(Median(arr1, arr2)); |
C#
using System; using System.Collections.Generic; namespace MedianOfTwoSortedArrays { class Program { static double Median(List< int > A, List< int > B) { int n = A.Count; int m = B.Count; int i; // initializing Priority Queue (Min Heap) var pq = new SortedSet< int >(); // pushing array A values to priority Queue for (i = 0; i < n; i++) pq.Add(A[i]); // pushing array B values to priority Queue for (i = 0; i < m; i++) pq.Add(B[i]); int check = n + m; double count = -1; double mid1, mid2; while (pq.Count != 0) { count++; // returning mid value if combined length(n+m) is odd if (check % 2 != 0 && count == check / 2) { double ans = pq.Min; return ans; } // maintaining mid1 value if combined length(n+m) is even // where we need to maintain both mid values in case of even combined length if (check % 2 == 0 && count == (check / 2) - 1){ mid1 = pq.Min; // now returning the mid2 value with previous maintained mid1 value by 2 if (check % 2 == 0 && count == check / 2) { mid2 = pq.Min; double ans = (mid1 + mid2) / 2; return ans; } } pq.Remove(pq.Min); } return 0.00000; } // Driver Code static void Main( string [] args) { List< int > arr1 = new List< int > { -2, 3, 4, 5 }; List< int > arr2 = new List< int > { -4, -1, 7, 8, 9 }; Console.WriteLine( "Median of the two arrays are" ); Console.WriteLine(Median(arr1, arr2)); Console.ReadLine(); } } } |
Median of the two arrays are 4
Time Complexity: O(max(N, M)*log(max(N, M))): Since the priority queue is implemented from two arrays
Auxiliary Space: O(N+M): for storing two array values in the priority queue.
Median of two sorted arrays of different sizes using Simulated stack:
Approach:
We create a new array with length that of the sum of the array lengths. We initialize i & j = 0. [i for nums1 & j for nums2]. Since the given arrays are already sorted it is easy to compare their elements. We comapre by observing nums1[i] < nums2[j] if the element in nums1nums1nums1 at ithi^{th}i the is less than that of element at jthj^{th}j th index of nums2nums2nums2, we add nums1[i]nums1[i]nums1[i] to new array and increment i; so as to compare the next element of the array to nums2[j].
If the opposite case arises, we add nums2[j]nums2[j]nums2[j] to the new array as you can guess. And increment j by 1 for the same reasons we did it with i.
Depending on the length of new array, we calculate median.
If the length of array is even, median by rule is the average of the 2 middle elements of the array
If it is off, it is the middlemost element
Intuition:
If n+m is odd, pop (n+m)//2 elements. Else pop (n+m)//2 -1. Write a pop function which compares the last values of both lists. Take into account the scenario where one of the listsis empty. Once the given number of elements are popped.If n+m was odd, next popped element is median. Else, mean of next 2 popped elements is median.
C++
#include <bits/stdc++.h> using namespace std; // Method to find median double findMedianSortedArrays(vector< int >& nums1, vector< int >& nums2) { int firstPtr = 0; // for interation on nums1 int secondPtr = 0; // for interation on nums2 int totalSize = nums1.size() + nums2.size(); // to check for odd position // median or even median int halfway = totalSize / 2 + 1; // total number of interation to run. // If total size is 6, at 4th index // of "merged" [we'll not actually // merge] we'll know the answer int last = numeric_limits< int >::min(); // simulate stack int secondLast = numeric_limits< int >::min(); // simulate stack int n = 0; // iteration count while (n < halfway && firstPtr < nums1.size() && secondPtr < nums2.size()) { if (nums1[firstPtr] <= nums2[secondPtr]) { n++; int val = nums1[firstPtr++]; // simulate push to stack secondLast = last; last = val; } else { n++; int val = nums2[secondPtr++]; // simulate push to stack secondLast = last; last = val; } } // if arr1 is bigger than arr2 while (n < halfway && firstPtr < nums1.size()) { n++; int val = nums1[firstPtr++]; // simulate push to stack secondLast = last; last = val; } // if arr2 is bigger than arr1 while (n < halfway && secondPtr < nums2.size()) { n++; int val = nums2[secondPtr++]; // simulate push to stack secondLast = last; last = val; } if (totalSize % 2 == 0) return (last + secondLast) / 2.0; return last; } // Driver code int main() { vector< int > arr1 = { -2, 3, 4, 5 }; vector< int > arr2 = { -4, -1, 7, 8, 9 }; cout << "Median of the two arrays are" << endl; cout << findMedianSortedArrays(arr1, arr2); return 0; } |
Median of the two arrays are 4
Time complexity: O(m+n), two array traversals simultaneously.
Space complexity: O(1), no extra space needed as it is a simulated stack.
Please Login to comment...