LC 1380.Lucky Numbers in a Matrix

题目描述

这是 LeetCode 上的 1380. 矩阵中的幸运数 ，难度为简单。

给你一个 m * n 的矩阵，矩阵中的数字 各不相同 。请你按 任意 顺序返回矩阵中的所有幸运数。

幸运数 是指矩阵中满足同时下列两个条件的元素：

• 在同一行的所有元素中最小
• 在同一列的所有元素中最大

示例 1：

输入：matrix = [[3,7,8],[9,11,13],[15,16,17]]
输出：[15]
解释：15 是唯一的幸运数，因为它是其所在行中的最小值，也是所在列中的最大值。

示例 2：

输入：matrix = [[1,10,4,2],[9,3,8,7],[15,16,17,12]]
输出：[12]
解释：12 是唯一的幸运数，因为它是其所在行中的最小值，也是所在列中的最大值。

示例 3：

输入：matrix = [[7,8],[1,2]]
输出：[7]
解释：7是唯一的幸运数字，因为它是行中的最小值，列中的最大值。

提示：

• m == mat.length
• n == mat[i].length
• 1 <= n, m <= 50
• 1 <= matrix[i][j] <= 10^5
• 矩阵中的所有元素都是不同的

解答

方法一：模拟

由于矩阵中的数都是不同的，所以我们遍历矩阵的每一行，先找出行的最小值，然后判断该最小值是否为同一列的最大值，如果是则添加到返回的数组中，如果不是则跳过。

class Solution {
    public List<Integer> luckyNumbers (int[][] matrix) {
        var res = new ArrayList<Integer>();
        for (int i = 0; i < matrix.length; ++i) {
            int row = 0;
            for (int j = 0; j < matrix[0].length; ++j) {
                if (matrix[i][row] > matrix[i][j]) {
                    row = j;
                }
            }
            boolean isMax = true;
            for (int j = 0; j < matrix.length; ++j) {
                if (matrix[i][row] < matrix[j][row]) {
                    isMax = false;
                    break;
                }
            }
            if (isMax) {
                res.add(matrix[i][row]);
            }
        }
        return res;
    }
}

• 时间复杂度：\(O(M(M + N))\)，其中 M 为矩阵的行数， N 为矩阵的列数。

• 空间复杂度：\(O(1)\)。

每题一图