add median of matrix (#9363)

* add median of matrix

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* update

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix formating

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Apply suggestions from code review

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Christian Clauss <cclauss@me.com>
This commit is contained in:
Bama Charan Chhandogi 2023-10-04 22:51:46 +05:30 committed by GitHub
parent d74349793b
commit 922d6a88b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

38
matrix/median_matrix.py Normal file
View File

@ -0,0 +1,38 @@
"""
https://en.wikipedia.org/wiki/Median
"""
def median(matrix: list[list[int]]) -> int:
"""
Calculate the median of a sorted matrix.
Args:
matrix: A 2D matrix of integers.
Returns:
The median value of the matrix.
Examples:
>>> matrix = [[1, 3, 5], [2, 6, 9], [3, 6, 9]]
>>> median(matrix)
5
>>> matrix = [[1, 2, 3], [4, 5, 6]]
>>> median(matrix)
3
"""
# Flatten the matrix into a sorted 1D list
linear = sorted(num for row in matrix for num in row)
# Calculate the middle index
mid = (len(linear) - 1) // 2
# Return the median
return linear[mid]
if __name__ == "__main__":
import doctest
doctest.testmod()