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;
        }
       
    }
};
相关推荐
thusloop1 小时前
380. O(1) 时间插入、删除和获取随机元素
数据结构·算法·leetcode
MobotStone2 小时前
无代码+AI时代,为什么你仍然需要像个开发者一样思考
人工智能·算法
緈福的街口2 小时前
【leetcode】584. 寻找用户推荐人
算法·leetcode·职场和发展
今天背单词了吗9802 小时前
算法学习笔记:17.蒙特卡洛算法 ——从原理到实战,涵盖 LeetCode 与考研 408 例题
java·笔记·考研·算法·蒙特卡洛算法
Maybyy3 小时前
力扣242.有效的字母异位词
java·javascript·leetcode
wjcurry3 小时前
完全和零一背包
数据结构·算法·leetcode
hie988943 小时前
采用最小二乘支持向量机(LSSVM)模型预测气象
算法·机器学习·支持向量机
python_tty4 小时前
排序算法(一):冒泡排序
数据结构·算法·排序算法
皮蛋sol周4 小时前
嵌入式学习C语言(八)二维数组及排序算法
c语言·学习·算法·排序算法
森焱森5 小时前
单片机中 main() 函数无 while 循环的后果及应对策略
c语言·单片机·算法·架构·无人机