目标
给你一个下标从 0 开始、大小为 m x n 的整数矩阵 matrix ，新建一个下标从 0 开始、名为 answer 的矩阵。使 answer 与 matrix 相等，接着将其中每个值为 -1 的元素替换为所在列的 最大 元素。
返回矩阵 answer 。
示例 1：
输入：matrix = [[1,2,-1],[4,-1,6],[7,8,9]]
输出：[[1,2,9],[4,8,6],[7,8,9]]
解释：上图显示了发生替换的元素（蓝色区域）。
- 将单元格 [1][1] 中的值替换为列 1 中的最大值 8 。
- 将单元格 [0][2] 中的值替换为列 2 中的最大值 9 。示例 2：
输入：matrix = [[3,-1],[5,2]]
输出：[[3,2],[5,2]]
解释：上图显示了发生替换的元素（蓝色区域）。说明：
- m == matrix.length
- n == matrix[i].length
- 2 <= m, n <= 50
-1 <= matrix[i][j] <= 100- 测试用例中生成的输入满足每列至少包含一个非负整数。
思路
使用矩阵各列的最大值替换该列中值为-1的元素。
先求出各列的最大值，然后替换。
代码
/**
* @date 2024-07-05 0:26
*/
public class ModifiedMatrix3033 {
/**
* 优化到一个循环中，使用局部遍历保存当前列的最大值，
* 没必要使用数组保存所有列的最大值然后再处理
*/
public int[][] modifiedMatrix_v1(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
for (int i = 0; i < n; i++) {
int max = 0;
for (int j = 0; j < m; j++) {
max = Math.max(max, matrix[j][i]);
}
for (int j = 0; j < m; j++) {
if (matrix[j][i] == -1) {
matrix[j][i] = max;
}
}
}
return matrix;
}
public int[][] modifiedMatrix(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
int[] max = new int[n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
max[i] = Math.max(max[i], matrix[j][i]);
}
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (matrix[j][i] == -1) {
matrix[j][i] = max[i];
}
}
}
return matrix;
}
}
性能
