LeetCode刷题---矩阵置零

2024-01-03 19:43:55

在这里插入图片描述
解题思路:

本题要求原地置换元素
对矩阵进行第一轮遍历,使用第一行第一列来充当该行该列是否要置换为0的标记位,如果第一行或第一列本身就含有零元素,我们使用colZero和rowZero变量来对其标记。如果第i行第j列的那个元素为0,我们就将[i][0]和[0][j]的元素置换为0,接着判断i和j是否为0,即第一行第一列,如果是,则colZero和rowZero都赋值为true。
对矩阵进行第二轮遍历:即对不包含第一行或第一列进行处理,根据第一轮遍历得到的标记将对应的行和列中的元素置换为0。
对矩阵进行第三轮遍历:即对第一行和第一列本身就含有0元素的情况进行处理,如果第一行或者第一列本身就包含0元素,则将其对应的行或列元素都置换为0。

代码实现:

 public void setZeroes(int[][] matrix) {
        int m=matrix.length;
        int n=matrix[0].length;
        boolean colZero=false;//第一行是否有0
        boolean rowZero=false;//第一列是否有0
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(matrix[i][j]==0)
                {
                    matrix[0][j]=matrix[i][0]=0;//标记位
                    //判断是否第一行第一列为0
                    if(i==0) rowZero=true;
                    if(j==0) colZero=true;
                }
            }
        }

        //第二轮遍历,将标记的行和列置换为0
        for(int i=1;i<m;i++)
        {
            for(int j=1;j<n;j++)
            {
                if(matrix[i][0]==0 || matrix[0][j]==0)
                {
                    matrix[i][j]=0;
                }
            }
        }

        //如果第一行或第一列本来就有0,则进行0置换
        for(int i=0; colZero && i<m;i++) matrix[i][0]=0;
        for(int j=0; rowZero && j<n;j++) matrix[0][j]=0;
    }

文章来源:https://blog.csdn.net/weixin_47109902/article/details/135369938
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。