leetcode每日一题-3033. 修改矩阵

题目描述:

解题思路:简单题目,思路非常直接。对列进行遍历,记录下最大值,然后再遍历一遍,把-1替换为最大值。需要注意的是进行列遍历和行遍历是不同的。

官方题解:

cpp 复制代码
class Solution {
public:
    vector<vector<int>> modifiedMatrix(vector<vector<int>>& matrix) {
        int n = matrix.size();
        int m = matrix[0].size();
        for (int j = 0; j < m; j++) {
            int zd = -1;
            for (int i = 0; i < n; i++) {
                zd = max(zd, matrix[i][j]);
            }
            for (int i = 0; i < n; i++) {
                if (matrix[i][j] == -1) {
                    matrix[i][j] = zd;
                }
            }
        }
        return matrix;
    }
};

其实还可以进一步优化。官方题解对数据进行了两次遍历,有没有办法进行一次遍历呢?事实上可以把每一列的最大值和-1的元素坐标先保存下来,然后再把元素坐标替换为相应列的最大值即可。

这样只需要遍历一次数据就可以了。

cpp 复制代码
class Solution {
public:
    // 存储坐标的结构体
    struct LOC{
        // 行坐标
        int row;
        // 列坐标
        int col;
    };
public:
    vector<vector<int>> modifiedMatrix(vector<vector<int>>& matrix) {
        // 获取行数
        int row = matrix.size();
        // 获取列数
        int col = matrix[0].size();
        // 没列最大值
        int max_value;
        // 存储每列最大值
        vector<int> max_vector;
        // 存储-1元素的坐标
        vector<LOC> loc_vector;
        // 按照列优先进行遍历
        for(int i=0;i<col; ++i)
        {
            // 假设最大值为-2
            max_value = -2;
            for(int j=0;j<row;++j)
            {
                if(max_value< matrix[j][i])
                    max_value=matrix[j][i];
                if(matrix[j][i]==-1)
                {
                    LOC tmp_loc;
                    tmp_loc.row = j;
                    tmp_loc.col = i;
                    loc_vector.push_back(tmp_loc);
                }
            }
            max_vector.push_back(max_value);
        }
        // 开始对-1的元素进行替换
        for(int i=0; i<loc_vector.size();++i)
        {
            matrix[loc_vector[i].row][loc_vector[i].col]=max_vector[loc_vector[i].col];
        }
        return matrix;

    }
};

这是一条吃饭博客,由挨踢零声赞助。学C/C++就找挨踢零声,加入挨踢零声,面试不挨踢!

相关推荐
雾月5530 分钟前
LeetCode 1292 元素和小于等于阈值的正方形的最大边长
java·数据结构·算法·leetcode·职场和发展
OpenC++1 小时前
【C++QT】Buttons 按钮控件详解
c++·经验分享·qt·leetcode·microsoft
知来者逆2 小时前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊2 小时前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北2 小时前
力扣-160.相交链表
算法·leetcode·链表
爱研究的小陈3 小时前
Day 3:数学基础回顾——线性代数与概率论在AI中的核心作用
算法
渭雨轻尘_学习计算机ing3 小时前
二叉树的最大宽度计算
算法·面试
BB_CC_DD4 小时前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘5 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV5 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法