力扣1380-矩阵中的幸运数-C++

一、题目描述

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

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

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

示例 1：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lucky-numbers-in-a-matrix

二、解题思路

首先逐行遍历，找到当前行的最小数，记录该最小数和所在列号，然后遍历该行最小数所在的列，找到该列的最大数，若行最小值和列最大值相等，则该数为幸运数。

三、代码

class Solution {
public:
    vector<int> luckyNumbers (vector<vector<int>>& matrix) {
        vector<int> res;
        int len1 = matrix.size();  //行数
        int len2 = matrix[0].size();  //列数
        if(len1 == 0 || len2 == 0) return res;
        int rowmin, colmax, col;  //col记录该行最小数的列号
        for(int i=0; i<len1; i++){
            rowmin = matrix[i][0];
            col = 0;  
            for(int j=1; j<len2; j++){  //寻找行最小数
                if(matrix[i][j] < rowmin){
                    rowmin = matrix[i][j];
                    col = j;
                }
            }
            colmax = matrix[0][col];
            for(int k=1; k<len1; k++){  //寻找列最小数
                if(matrix[k][col] > colmax){
                    colmax = matrix[k][col];
                }
            }
            if(rowmin == colmax){  //该数为幸运数
                res.push_back(rowmin);
            }
        }
        return res;
    }
};

四、运行结果