Python Program to Find Median of Two Sorted Arrays of Equal Size
Median of two sorted arrays of equal size in Python
Here, on this page, we will discuss the program to find the Median of two sorted arrays of equal size in the Python programming language. We are given two arrays say arr1[ ] and arr2[ ] of the same size say n. We need to find the median after merging these arrays.
Example :
- Input : arr1 = [ 1, 12, 15, 26, 38 ], arr2 = [ 2, 13, 17, 30, 45 ]
- Output : 16
- Explanation : After merging and sorting the array it will be [ 1, 2, 12, 13, 15, 17, 26, 30, 38, 45 ], middle elements are 15 and 17 so, median of the given array will be ((15+17)/2) = 16
Algorithm
- After merging both arrays and sorting them send it to the function to find median
- If the length of the array is even then find the floor value of l divided by 2 and store it in medium
- Set median equals to sum of (array[medium] and array[medium-1]) divided by 2
- return median
- Else set medium equals to ceil(l/2) and return array[medium -1]
Python Code
Run
from math import ceil
def median_of_two_arrays(l, array):
if (l % 2) == 0:
medium = l // 2
median = (array[medium] + array[medium - 1]) / 2
return median
else:
medium = ceil(l / 2)
return array[medium - 1]
array1 = [1, 12, 15, 26, 38]
array2 = [2, 13, 17, 30, 45]
print("Array after merging and sorting :", end="")
array1 += array2
array1.sort()
print(array1)
ans = median_of_two_arrays(len(array1), array1)
print("Median :", int(ans))
Output
Array after merging and sorting :[1, 2, 12, 13, 15, 17, 26, 30, 38, 45]
Median : 16
Prime Course Trailer
Related Banners
Get PrepInsta Prime & get Access to all 200+ courses offered by PrepInsta in One Subscription
For similar Questions click on the given button.
Login/Signup to comment