leetcode刷题记录-1380. 矩阵中的幸运数

510 阅读4分钟

「这是我参与2022首次更文挑战的第20天,活动详情查看:2022首次更文挑战

前言

今日的题目为简单,认真读懂题目的意思由于数据范围很小,所以哪怕是用最笨的方法也能够实现,由于题目的限制比较大,所以最后的答案还能的得出一点可以优化的地方。

每日一题

今天的每日一题 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 == mat.length
  • n == mat[i].length
  • 1 <= n, m <= 50
  • 1 <= matrix[i][j] <= 10^5
  • 矩阵中的所有元素都是不同的

题解

第一种题解 预处理

本道题为简单题,因为题目给的数据范围较小 1 <= n, m <= 50 ,并且还限制了所有的元素都是不同的。

我们可以先双重遍历一遍数组,建立两个保存行最小值和列最大值的数组,然后再遍历一遍数组,当碰到一个数即在最小值数组中又在最大值数组中,就是答案,将其加入数组。

  1. 先双重循环遍历输入的二维数组,得到最大列表和最小列表:
 let minRow = new Array(n).fill(Number.MAX_SAFE_INTEGER);
let maxCol = new Array(m).fill(0);

for (let i = 0; i < n; i++) {
    for (let j = 0; j < m; j++) {
        minRow[i] = Math.min(minRow[i], matrix[i][j]);
        maxCol[j] = Math.max(maxCol[j], matrix[i][j]);
    }
}
  1. 然后在遍历一次找到即在最小列表又在最大列表的那个:
let ans = [];
   
for (let i = 0; i < n; i++) {
    for (let j = 0; j < m; j++) {
        if (matrix[i][j] === minRow[i] && matrix[i][j] === maxCol[j]) {
            ans.push(matrix[i][j]);
        }
    }
}

最后的ans数组就是题目需要的答案。

/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var luckyNumbers  = function(matrix) {
    let n = matrix.length
    let m = matrix[0].length
    let ans = [];
    let minRow = new Array(n).fill(Number.MAX_SAFE_INTEGER);
    let maxCol = new Array(m).fill(0);

    for (let i = 0; i < n; i++) {
        for (let j = 0; j < m; j++) {
            minRow[i] = Math.min(minRow[i], matrix[i][j]);
            maxCol[j] = Math.max(maxCol[j], matrix[i][j]);
        }
    }
    for (let i = 0; i < n; i++) {
        for (let j = 0; j < m; j++) {
            if (matrix[i][j] === minRow[i] && matrix[i][j] === maxCol[j]) {
                ans.push(matrix[i][j]);
            }
        }
    }
    return ans;
};

image.png

第二种题解 哈希表保存状态

由于题目限制了所有的元素都是不同的。

所以我们可以直接按照题目的要求,找到每一行中最小的数,再去找到每一行中最大的数,由于数据不会重复,所以二者中相同的那个数就是我们要找的答案,没有就是不存在。

优化点

这里有一个需要注意到的点,一个数在行中最小,列中最大,这样的数,是不可能有两个的。如下图:

image.png

我们先假设有两个幸运数,分别是 ai

a 在行中最小,在列中最大,那么可以得到 g < a < c

i 在行中最小,在列中最大,那么可以得到 c < i < g

那么这四个数就死循环了:

image.png

很明显这是不可能实现的,那么幸运数就一定只有一个

解题过程

  1. 我们先拿到每列最大的数,然后将他们都存入一个哈希表中。

  2. 我们再去拿到每一行最小的数和哈希表做一个判断,如果最小的数出现在了哈希表,注意,这时候我们就可以直接把这个数返回出去,因为上面的分析让我们知道了这个数一定只有一个。

  3. 然后要记得做一个没有找到的判断,找不到这个数就返回一个空数组。

/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var luckyNumbers  = function(matrix) {
let n = matrix.length
    let m = matrix[0].length
    let map = new Map()
    let max = 0;
    for (let j = 0; j < m; j++) {
        for(let i = 0; i < n; i++){
            if(max < matrix[i][j]){
                max = matrix[i][j]
            }
        }
        map.set(max, 1)
        max = 0;
    }
    for (let i = 0; i < n; i++) {
        let min = Math.min(...matrix[i])
        if(map.has(min)){
            return [min]
        }
    }
    return []
};

image.png