开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情
一、题目描述:
给你一个 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;
}