mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-27 23:11:09 +00:00
34 lines
978 B
Python
34 lines
978 B
Python
|
from typing import List
|
||
|
|
|
||
|
|
|
||
|
|
def median_of_two_arrays(nums1: List[float], nums2: List[float]) -> float:
|
||
|
|
"""
|
||
|
|
>>> median_of_two_arrays([1, 2], [3])
|
||
|
|
2
|
||
|
|
>>> median_of_two_arrays([0, -1.1], [2.5, 1])
|
||
|
|
0.5
|
||
|
|
>>> median_of_two_arrays([], [2.5, 1])
|
||
|
|
1.75
|
||
|
|
>>> median_of_two_arrays([], [0])
|
||
|
|
0
|
||
|
|
>>> median_of_two_arrays([], [])
|
||
|
|
Traceback (most recent call last):
|
||
|
|
...
|
||
|
|
IndexError: list index out of range
|
||
|
|
"""
|
||
|
|
all_numbers = sorted(nums1 + nums2)
|
||
|
|
div, mod = divmod(len(all_numbers), 2)
|
||
|
|
if mod == 1:
|
||
|
|
return all_numbers[div]
|
||
|
|
else:
|
||
|
|
return (all_numbers[div] + all_numbers[div - 1]) / 2
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import doctest
|
||
|
|
|
||
|
|
doctest.testmod()
|
||
|
|
array_1 = [float(x) for x in input("Enter the elements of first array: ").split()]
|
||
|
|
array_2 = [float(x) for x in input("Enter the elements of second array: ").split()]
|
||
|
|
print(f"The median of two arrays is: {median_of_two_arrays(array_1, array_2)}")
|