题目
给你一个 m * n 的矩阵,矩阵中的数字 各不相同 。请你按 任意 顺序返回矩阵中的所有幸运数。
幸运数是指矩阵中满足同时下列两个条件的元素:
在同一行的所有元素中最小
在同一列的所有元素中最大
示例:
输入:matrix = [[3,7,8],[9,11,13],[15,16,17]]
输出:[15]
解释:15 是唯一的幸运数,因为它是其所在行中的最小值,也是所在列中的最大值。
分析
幸运数是同一行的最小,同一列的最大,那么我们可以对每一行进行升序排序,对每一列进行降序排序
暴力解法
我们将每一行的最小值加入list,将每一列的最大值加入list2,然后遍历list,看list是否有list2中的元素,如果有则加入一个新的list里面,最后返回list。
public List<Integer> luckyNumbers(int[][] matrix) {
List<Integer> list1 = new ArrayList<>();
List<Integer> list2 = new ArrayList<>();
// 找出每一行的最小值
for (int i = 0; i < matrix.length; i++) {
int tempRowMin = Integer.MAX_VALUE;
for (int j = 0; j < matrix[i].length; j++) {
if (tempRowMin > matrix[i][j]) {
tempRowMin = matrix[i][j];
}
}
list1.add(tempRowMin);
}
// 找出每一列的最大值
for (int i = 0; i < matrix[i].length; i++) {
int tempColMax = Integer.MIN_VALUE;
for (int j = 0; j < matrix.length; j++) {
if (tempColMax < matrix[j][i]) {
tempColMax = matrix[j][i];
}
}
list2.add(tempColMax);
}
List<Integer> list = new ArrayList<>();
for (Integer temp : list1) {
if (list2.contains(temp)) {
list.add(temp);
}
}
return list;
}
还有一种思路:先找每一行的最小数,然后去判断,在它所在的那一列是否有比它大的数,如果没有说明为幸运数,如果有则返回false;
class Solution {
public List<Integer> luckyNumbers(int[][] matrix) {
int row = matrix.length;
int col = matrix[0].length;
List<Integer> ans = new ArrayList<>();
//遍历每一行
for (int i = 0; i < row; i++) {
//找行最小值
int loc = find(matrix, i);
//判断是否有比它大的数
if (isMaxOfCol(matrix, i, loc)) {
//如果没有则为幸运数
ans.add(matrix[i][loc]);
}
}
return ans;
}
// 找每一行的最小值
public int find(int[][] matrix, int row) {
int minOfRow = Integer.MAX_VALUE;
int loc = 0;//记录位置
for (int i = 0; i < matrix[row].length; i++) {
if (minOfRow > matrix[row][i]) {
minOfRow = matrix[row][i];
loc = i;
}
}
return loc;
}
// 判断是否是每一列的最大值
public boolean isMaxOfCol(int[][] matrix, int row, int col) {
//传入的col不变,为find()函数找到的行最小值
for (int i = 0; i < matrix.length; i++) {
if (row == i) {
continue;
}
//进行比较,如果对应列的其余行有比它大的,则不是幸运数,否则就是
if (matrix[i][col] > matrix[row][col]) {
return false;
}
}
return true;
}
}
反思
这道题不难,就是挺麻烦的,有点绕
end.