Add median of matrix (#4590)

This commit is contained in:
Bama Charan Chhandogi
2023-10-03 19:46:14 +05:30
committed by GitHub
parent 536978919d
commit 5f5a61de87
2 changed files with 64 additions and 0 deletions

View File

@ -0,0 +1,30 @@
package com.thealgorithms.misc;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
/**
* Median of Matrix (https://medium.com/@vaibhav.yadav8101/median-in-a-row-wise-sorted-matrix-901737f3e116)
* Author: Bama Charan Chhandogi (https://github.com/BamaCharanChhandogi)
*/
public final class MedianOfMatrix {
public static int median(List<List<Integer>> matrix) {
// Flatten the matrix into a 1D list
List<Integer> linear = new ArrayList<>();
for (List<Integer> row : matrix) {
linear.addAll(row);
}
// Sort the 1D list
Collections.sort(linear);
// Calculate the middle index
int mid = (0 + linear.size() - 1) / 2;
// Return the median
return linear.get(mid);
}
}