跳至主要內容

1380. 矩阵中的幸运数


1380. 矩阵中的幸运数

🟢   🔖  数组 矩阵  🔗 力扣open in new window LeetCodeopen in new window

题目

Given an m x n matrix of distinct numbers, return alllucky numbers in the matrix in any order.

A lucky number is an element of the matrix such that it is the minimum element in its row and maximum in its column.

Example 1:

Input: matrix = [[3,7,8],[9,11,13],[15,16,17]]

Output: [15]

Explanation: 15 is the only lucky number since it is the minimum in its row and the maximum in its column.

Example 2:

Input: matrix = [[1,10,4,2],[9,3,8,7],[15,16,17,12]]

Output: [12]

Explanation: 12 is the only lucky number since it is the minimum in its row and the maximum in its column.

Example 3:

Input: matrix = [[7,8],[1,2]]

Output: [7]

Explanation: 7 is the only lucky number since it is the minimum in its row and the maximum in its column.

Constraints:

  • m == mat.length
  • n == mat[i].length
  • 1 <= n, m <= 50
  • 1 <= matrix[i][j] <= 10^5.
  • All elements in the matrix are distinct.

题目大意

给你一个 m x 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. 假设有两个幸运数 X 和 Y

      c2              c1
      |               |
r1 ---+-------------- X ---
      |               |
      |               |
      |               |
r2 -- Y --------------+----
      |               |

假设矩阵中有两个幸运数 XY,它们分别位于不同的位置:

  • X 位于第 r1 行第 c1 列。
  • Y 位于第 r2 行第 c2 列。

根据题目定义,XY 都是幸运数,意味着:

  • X 是第 r1 行的最小值,并且是第 c1 列的最大值。
  • Y 是第 r2 行的最小值,并且是第 c2 列的最大值。

2. Y < X

      c2             c1
      |               |
r1 ---+-------------- X ---
      |               |
      |               |
      |               |
r2 -- Y ------------- A ---
      |               |

假设 A 点位于第 r2 行第 c1 列,则可以得到:

  • X > A,因为 X 是第 c1 列的最大值。
  • Y < A,因为 Y 是第 r2 行的最小值。

因此可以得到:Y < X

3. Y > X

      c2             c1
      |               |
r1 -- B ------------- X ---
      |               |
      |               |
      |               |
r2 -- Y ------------- A ---
      |               |

假设 B 点位于第 r1 行第 c2 列,则可以得到:

  • X < B,因为 X 是第 r1 行的最小值。
  • Y > B,因为 Y 是第 c2 列的最大值。

因此可以得到:Y > X

4. 得出矛盾

从上面的描述可以看出,如果 XY 都是幸运数,当它们位于不同的行和列时,那么可以推出 Y < XY > X,这两个结论相互冲突,即不可能同时存在两个幸运数。

通过反证法,我们得出结论:矩阵中至多只有一个幸运数

找出幸运数

  • 幸运数字是所在行的最小值,又是所在列的最大值。
  • 首先,找到每行的最小元素,然后将这些最小值中的最大值确定为 rowMinMax
  • 同样,找到每列的最大值,然后将这些最大值中的最小值确定为 colMaxMin
  • 如果 rowMinMax 等于 colMaxMin,则该值就是幸运数字;
  • 否则,返回一个空数组。

复杂度分析

  • 时间复杂度O(m * n),其中 m 是行数,n 是列数,遍历每一行和每一列各一次。
  • 空间复杂度O(1),只使用了常数空间来存储变量

代码

/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var luckyNumbers = function (matrix) {
	const m = matrix.length;
	const n = matrix[0].length;

	// 找到每行的最小值
	let rowMinMax = -Infinity;
	for (let i = 0; i < m; i++) {
		let rowMin = Infinity;
		for (let j = 0; j < n; j++) {
			rowMin = Math.min(rowMin, matrix[i][j]);
		}
		rowMinMax = Math.max(rowMinMax, rowMin);
	}

	// 找到每列的最大值
	let colMaxMin = Infinity;
	for (let i = 0; i < n; i++) {
		let colMax = -Infinity;
		for (let j = 0; j < m; j++) {
			colMax = Math.max(colMax, matrix[j][i]);
		}
		colMaxMin = Math.min(colMaxMin, colMax);
	}

	// 判断是否为幸运数
	if (rowMinMax === colMaxMin) {
		return [rowMinMax];
	}

	return [];
};