leetcode 1380. 矩阵中的幸运数

2022-02-15

题目

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

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

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

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

难度：Easy

思路

简单模拟即可。可以预处理一下矩阵，记录每一行的最小值和每一列的最大值。再逐个元素检查

技能点：模拟

代码

class Solution {
public:
    vector<int> luckyNumbers (vector<vector<int>>& matrix) {
        vector<int> ans;
        int n = matrix.size(), m = matrix[0].size();
        vector<int> row(n, INT_MAX), col(m, 0);
        for(int i=0;i<n;++i){
            for(int j=0;j<m;++j){
                row[i] = min(row[i], matrix[i][j]);
                col[j] = max(col[j], matrix[i][j]);
            }
        }
        for(int i=0;i<n;++i){
            for(int j=0;j<m;++j){
                if(matrix[i][j] == row[i] && matrix[i][j] == col[j]){
                    ans.emplace_back(matrix[i][j]);
                }
            }
        }
        return ans;
    }
};