[LeetCode 周赛180] 1. 矩阵中的幸运数(暴力、代码优化)

1. 题目来源

链接：1380. 矩阵中的幸运数

2. 题目说明

3. 题目解析

方法一：暴力+常规解法

题意很明确，暴力枚举即可。看了看评论+题解区的大佬也都是暴力求解的，话说 py 三行党是真的…简单说下思路：

• 首先对列进行遍历，求出列的最大元素，再取出列最大元素那一行调用 API 排序，找到最小的，因为元素不重复，所以当最大数等于最小数即为幸运数，添加到 vector 即可

很不优雅的写法，想到了就直接码上了…

参见代码如下：

// 执行用时 :80 ms, 在所有 C++ 提交中击败了100.00%的用户
// 内存消耗 :11.8 MB, 在所有 C++ 提交中击败了100.00%的用户

class Solution {
public:
    vector<int> luckyNumbers (vector<vector<int>>& matrix) {
        int row = matrix.size(); 
        int col = matrix[0].size();
        vector<int> vt;
        int cnt = 0;
        for (int i = 0; i < col; ++i) {
            int max = -11111111;
            int min = 999999999;
            for (int j = 0; j < row; ++j) {
                if (max < matrix[j][i]) {
                    vector<int> tmp(matrix[j]);
                    sort(tmp.begin(), tmp.end());
                    max = matrix[j][i];
                    cnt = tmp[0];
                }
            }
            if (max == cnt) {
                vt.push_back(max);
            }
        }
        return vt;
    }
};

方法二：暴力+函数+常规解法

发现自己在短时间还是无法写出优雅的解法，基础还是不牢固啊。简单总结下下面的思路：

• 暴力遍历矩阵
• 需要判断函数，那就写一个判断函数 check 若当前位置不是行最小、列最大直接返回 false 其自然也不为幸运数
• 若满足 check 为幸运数则将其添加到 vt 中

参见代码如下：

// 执行用时 :28 ms, 在所有 C++ 提交中击败了100.00%的用户
// 内存消耗 :9.6 MB, 在所有 C++ 提交中击败了100.00%的用户

class Solution {
public:
    bool check(int n, int m, int x, int y, vector<vector<int>>& matrix){
        for (int i = 0; i < n; i++) {
            if (matrix[i][y] > matrix[x][y]) return false;
        }
        for (int j = 0; j < m; j++) {
            if (matrix[x][j] < matrix[x][y]) return false;
        }
        return true;
    }
    
    vector<int> luckyNumbers (vector<vector<int>>& matrix) {
        int n = matrix.size(), m = matrix[0].size();
        vector<int> vt;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++){
                if (check(n, m, i, j, matrix)) vt.push_back(matrix[i][j]);
            }
        }
        return vt;
    }
};