行排序矩阵的中位数

/**
 * LeetCode 2387: Row Sorted Matrix Median
 *
 * 思路:
 * 1. 对于每一行，找到其中位数。
 * 2. 将所有行的中位数存储在一个列表中。
 * 3. 对于存储了所有行中位数的列表，找到其总体的中位数。
 */

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class SolutionStream {
    public int findMedian(int[][] matrix) {
        List<Integer> medians = Arrays.stream(matrix)
                .map(row -> findRowMedian(row))
                .sorted()
                .collect(Collectors.toList());
        return medians.get(medians.size() / 2);
    }

    private int findRowMedian(int[] row) {
        int n = row.length;
        return (n % 2 == 1) ? row[n / 2] : (row[(n / 2) - 1] + row[n / 2]) / 2;
    }

    public static void main(String[] args) {
        SolutionStream solution = new SolutionStream();
        int[][] matrix = {{1, 3, 5}, {2, 6, 9}, {3, 6, 9}};
        System.out.println(solution.findMedian(matrix)); // 应输出6
    }
}