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

106 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情

一、题目描述:

1380. 矩阵中的幸运数 - 力扣(LeetCode)

给你一个 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
  • 矩阵中的所有元素都是不同的

二、思路分析:

这题是赤裸裸的模拟题。有两种思路可供参考。

##思路1

先求出每一行的最小值所在的位置,然后对于每一行的最小值,判断其是否为对应的列的最大值。

##思路2

针对每一个元素 matrix[i][j] ,判断其是否为所在行的最小值以及所在列的最大值。

如果是,则更新行最小值所在位置 ( rowmin[i] )/列最大值所在位置( colmax[j] )。

接着再对每一个元素进行遍历,如果 matrix[i][j] == rowmin[i] && matrix[i][j] == colmax[j] 则符合题意。

三、AC 代码:

vector<int> luckyNumbers(vector<vector<int>> &matrix)
{
    int m = matrix.size();
    int n = matrix[0].size();
    vector<int> rmin(m, INT_MAX);
    vector<int> cmax(n, INT_MIN);
    vector<int> ans;
    for (int i = 0; i < m; i++)
    {
        for (int j = 0; j < n; j++)
        {
            rmin[i] = min(rmin[i], matrix[i][j]);
            cmax[j] = max(cmax[j], matrix[i][j]);
        }
    }
    for (int i = 0; i < m; i++)
    {
        for (int j = 0; j < n; j++)
        {
            if (matrix[i][j] == rmin[i] && matrix[i][j] == cmax[j])
                ans.push_back(matrix[i][j]);
        }
    }
    return ans;
}

四、参考:

借助2个vector,减少判断次数,期待后面继续优化 - 矩阵中的幸运数 - 力扣(LeetCode)s