题目
给你一个 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
矩阵中的所有元素都是不同的
解释与代码
用map来存i和j,如果相同把遍历到的数字push_back进ans,模拟即可
还有一种预处理的方式,即确定每一行和每一列的最大最小值,扫描的时候如果两个数字相同即为幸运数
class Solution {
public:
vector<int> luckyNumbers (vector<vector<int>>& matrix) {
vector<int> ans;
map<int,int> mii;
int dx = 0x3f3f3f3f, cnt = 0x3f3f3f3f;
for (int i=0; i<matrix.size(); i++) {
dx = 0x3f3f3f3f, cnt = 0x3f3f3f3f;
for (int j=0; j<matrix[0].size(); j++) {
if (matrix[i][j] < dx) {
dx = matrix[i][j];
cnt = j;
}
}
mii[i] = cnt;
}
for (int j=0; j<matrix[0].size(); j++) {
dx = -1, cnt = -1;
for (int i=0; i<matrix.size(); i++) {
if (matrix[i][j] > dx) {
dx = matrix[i][j];
cnt = i;
}
}
if (mii[cnt] == j) ans.push_back(matrix[cnt][j]);
}
return ans;
}
};