1380. 矩阵中的幸运数

624 阅读1分钟

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]

提示:

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

解题思路

解法一: 按幸运数定义查找

  1. 循环matrix矩阵,获取每行中最小的元素
  2. 判断第一步获取的元素是否为所在列最大元素
/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var luckyNumbers  = function(matrix) {
    // 按幸运数定义查找
    let m = matrix.length, n = matrix[0].length, lucky = []
    // 循环 matrix 矩阵
    for (let i = 0; i < m; i++) {
        // 找到本行最小的元素, 记录该元素的值与所在列索引
        let min = matrix[i][0], ind = 0
        for (let j = 1; j < n; j++) {
            if (matrix[i][j] < min) {
                min = matrix[i][j]
                ind = j
            }
        }
        // 判断是否为所在列的最大元素
        for (let j = 0; j < m; j++) {
            // 存在更大值, 终止
            if (matrix[j][ind] > min) break
            // 满足条件加入 lucky
            if (j == m - 1) lucky.push(min)
        }
    }
    return lucky
};

解法二: 获取最小值与最大值集合

  1. 获取每行元素中的最小值
  2. 获取每列元素中的最大值
  3. 查找最小值与最大值集合的交集
/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var luckyNumbers  = function(matrix) {
    // 获取每行最小值集合
    let mins = matrix.map(row => Math.min(...row))
    // 获取每列最大值集合
    let maxs = matrix[0].map((item, col_index) => Math.max(...matrix.map(row => row[col_index])))
    // 最小值与最大值集合的交集即为幸运数
    return maxs.filter(item => mins.includes(item))
};