3033. 修改矩阵

题目描述：

给你一个下标从 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 元素替换为该列的最大元素

func modifiedMatrix(matrix [][]int) [][]int {
	var m, n = len(matrix), len(matrix[0])
	var max = make([]int, n)
	for i := 0; i < n; i++ {
		for j := 0; j < m; j++ {
			if matrix[j][i] >= max[i] {
				max[i] = matrix[j][i]
			}
		}
	}
	for i := 0; i < n; i++ {
		for j := 0; j < m; j++ {
			if matrix[j][i] == -1 {
				matrix[j][i] = max[i]
			}
		}
	}
	return matrix
}

复杂度：

• 时间复杂度：O(N * M)
• 空间复杂度：O(N)

执行结果：

• 执行耗时:9 ms,击败了30.00% 的Go用户
• 内存消耗:4 MB,击败了43.33% 的Go用户