修改矩阵

2024-07-05

题目:

给你一个下标从 0 开始、大小为 m x n 的整数矩阵 matrix ,新建一个下标从 0 开始、名为 answer 的矩阵。使 answermatrix 相等,接着将其中每个值为 -1 的元素替换为所在列的 最大 元素。返回矩阵 answer

示例 1:

img

输入: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:

img

输入: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
  • 测试用例中生成的输入满足每列至少包含一个非负整数。

思路:

简单题,不太需要思路

代码:

class Solution {
public:
    vector<vector<int>> modifiedMatrix(vector<vector<int>>& matrix) {
        int m = matrix.size(), n = matrix[0].size(); 
        vector<vector<int>> answer(matrix);
        for (int j = 0; j < n; j++) {
            int maxVal = -2;
            vector<int> negIdx;
            for (int i = 0; i < m; i++) {
                if (matrix[i][j] == -1) {
                    negIdx.push_back(i);
                }
                maxVal = max(maxVal, matrix[i][j]);
            }
            for (int k = 0; k < negIdx.size(); k++) {
                answer[negIdx[k]][j] = maxVal;
            }
        }
        return answer;   
    }
};