LeetCode 73. 矩阵置零

题目链接

力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台

题目解析

1、定义两个变量,flag_col0与flag_row0。分别代表第一列和第一行是否为0。我们只需要遍历第一行和第一列,只要遇见0则对应的第一行和第一列就赋值为true.

2、我们使用第一行和第一列的数来去记录该行是否出现0,和该列是否出现0.若某行出现0,则该数所对应的第一行与第一列的数设为0.

3、根据第一行与第一列的情况来对整个数组(除了第一行和第一列)进行赋值。

4、根据flag_col0与flag_row0来去对第一行和第一列进行赋值。

代码

cpp 复制代码
class Solution 
{
public:
    void setZeroes(vector<vector<int>>& matrix)
    {
        int m=matrix.size();
        int n=matrix[0].size();
        // flag_col0 记录第一列是否出现0 也就是第一列是否需要全部赋值为0(只要有一个为0最终都为0)
        // flag_row0 记录第一行是否出现0 也就是第一行是否需要全部赋值为0(只要有一个为0最终都为0)
        // 第一列用来记录每行是否出现0 
        // 第一行用来记录每列是否出现0
        // 最终使用flag_col0与flag_row0重新对第一列的数据和第一行的数据进行赋值

        int flag_col0 = false, flag_row0 = false;
        // 判断flag_col0
        for(int i=0;i<m;i++)
        {
            if(!matrix[i][0]) flag_col0=true;
        }
        // 判断flag_row0
        for(int j=0;j<n;j++)
        {
            if(!matrix[0][j]) flag_row0=true;
        }
        // 遍历每个元素,若该元素为0,matrix[i][0]=matrix[0][j]=0
        for(int i = 1; i < m; i++)
            for(int j = 1; j<n;j++)
            {
                if(!matrix[i][j])
                {
                    matrix[i][0]=matrix[0][j]=0;
                }
            }
        // 根据我们得到的第一行与第一列是否为0,来对数据重新赋值
        for(int i = 1; i < m; i++)
            for(int j = 1; j<n;j++)
            {
                if(!matrix[i][0]||!matrix[0][j])
                {
                    matrix[i][j]=0;
                }
            }
        // 对第一行和第一列重新赋值
        if(flag_col0)
        {
            for(int i=0;i<m;i++)
                matrix[i][0]=0;
        }
        if(flag_row0)
        {
            for(int i=0;i<n;i++)
             matrix[0][i]=0;
        }
       
    }
};
相关推荐
董董灿是个攻城狮1 小时前
5分钟搞懂什么是窗口注意力?
算法
Dann Hiroaki1 小时前
笔记分享: 哈尔滨工业大学CS31002编译原理——02. 语法分析
笔记·算法
qqxhb3 小时前
零基础数据结构与算法——第四章:基础算法-排序(上)
java·数据结构·算法·冒泡·插入·选择
FirstFrost --sy5 小时前
数据结构之二叉树
c语言·数据结构·c++·算法·链表·深度优先·广度优先
森焱森5 小时前
垂起固定翼无人机介绍
c语言·单片机·算法·架构·无人机
搂鱼1145145 小时前
(倍增)洛谷 P1613 跑路/P4155 国旗计划
算法
Yingye Zhu(HPXXZYY)5 小时前
Codeforces 2021 C Those Who Are With Us
数据结构·c++·算法
无聊的小坏坏6 小时前
三种方法详解最长回文子串问题
c++·算法·回文串
长路 ㅤ   7 小时前
Java后端技术博客汇总文档
分布式·算法·技术分享·编程学习·java后端
秋说7 小时前
【PTA数据结构 | C语言版】两枚硬币
c语言·数据结构·算法