题目描述
给你一个 m * n
的矩阵,矩阵中的数字 各不相同 。请你按 任意 顺序返回矩阵中的所有幸运数。
幸运数是指矩阵中满足同时下列两个条件的元素:
-
在同一行的所有元素中最小
-
在同一列的所有元素中最大
样例
输入:matrix = [[3,7,8],[9,11,13],[15,16,17]]
输出:[15]
解释:15 是唯一的幸运数,因为它是其所在行中的最小值,也是所在列中的最大值。
输入:matrix = [[1,10,4,2],[9,3,8,7],[15,16,17,12]]
输出:[12]
解释:12 是唯一的幸运数,因为它是其所在行中的最小值,也是所在列中的最大值。
输入:matrix = [[7,8],[1,2]]
输出:[7]
提示:
-
m == mat.length
-
n == mat[i].length
-
1 <= n, m <= 50
-
1 <= matrix[i][j] <= 10^5
矩阵中的所有元素都是不同的
算法分析
- 1、将每一行的最小值,每一列的最大值预处理出来
- 2、枚举所有位置的元素,若位置的元素是当前行最小且当前列最大,则加到链表中
时间复杂度$O(nm)$
Java 代码
class Solution {
static int INF = 0x3f3f3f3f;
public List<Integer> luckyNumbers (int[][] matrix) {
List<Integer> ans = new ArrayList<Integer>();
int n = matrix.length;
int m = matrix[0].length;
int[] row = new int[n + 10];
int[] col = new int[m + 10];
Arrays.fill(col,-INF);
for(int i = 0;i < n;i ++)
{
int res = INF;
for(int j = 0;j < m;j ++)
{
col[j] = Math.max(col[j],matrix[i][j]);
res = Math.min(res,matrix[i][j]);
}
row[i] = res;
}
for(int i = 0;i < n;i ++)
{
for(int j = 0;j < m;j ++)
{
if(matrix[i][j] == row[i] && row[i] == col[j]) ans.add(row[i]);
}
}
return ans;
}
}