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
- 矩阵中的所有元素都是不同的
解题思路
解法一: 按幸运数定义查找
- 循环matrix矩阵,获取每行中最小的元素
- 判断第一步获取的元素是否为所在列最大元素
/**
* @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
};
解法二: 获取最小值与最大值集合
- 获取每行元素中的最小值
- 获取每列元素中的最大值
- 查找最小值与最大值集合的交集
/**
* @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))
};